AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-372153 - Add hipStreamGetDevice Implementation

Browse Source 
Change-Id: Ifd1f13e311e8221ca6d94cf27f9131eb97678067
This commit is contained in:
Jatin Chaudhary
2023-01-17 10:40:05 +00:00
committed by Jatin Jaikishan Chaudhary
parent 3fc34b8bf5
commit 8ce9ec60f8
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
+14
View File
@@ -2507,6 +2507,20 @@ inline static hipError_t hipStreamAddCallback(hipStream_t stream, hipStreamCallb
cudaStreamAddCallback(stream, (cudaStreamCallback_t)callback, userData, flags));
}
inline static hipError_t hipStreamGetDevice(hipStream_t stream, hipDevice_t* device) {
hipCtx_t context;
auto err = hipCUResultTohipError(cuStreamGetCtx(stream, &context));
if (err != hipSuccess) return err;
err = hipCUResultTohipError(cuCtxPushCurrent(context));
if (err != hipSuccess) return err;
err = hipCUResultTohipError(cuCtxGetDevice(device));
if (err != hipSuccess) return err;
return hipCUResultTohipError(cuCtxPopCurrent(&context));
}
inline static hipError_t hipDriverGetVersion(int* driverVersion) {
return hipCUDAErrorTohipError(cudaDriverGetVersion(driverVersion));
}