mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #7315 from ROCmSoftwarePlatform:fix_pr_7306_rocm

Browse Source 
PiperOrigin-RevId: 385566677
This commit is contained in:
jax authors 2021-07-19 09:01:05 -07:00
commit 29187a3317
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/rocblas.cc
View File

@ -67,7 +67,7 @@ template <>
if (stream) {
ThrowIfErrorStatus(rocblas_set_stream(handle, stream));
}
return rocBlasHandlePool::Handle(pool, handle);
return rocBlasHandlePool::Handle(pool, handle, stream);
}
// Set of types known to Rocsolver.