[cuBLAS] Fix deprecated API (uxlfoundation#95)

mmeterel · web-flow · commit 58fc2c6fd9b2 · 2021-05-20T10:33:50.000-07:00
diff --git a/src/blas/backends/cublas/cublas_scope_handle.cpp b/src/blas/backends/cublas/cublas_scope_handle.cpp
@@ -97,7 +97,8 @@ void ContextCallback(void *userData) {
 }
 
 cublasHandle_t CublasScopedContextHandler::get_handle(const cl::sycl::queue &queue) {
-    auto piPlacedContext_ = reinterpret_cast<pi_context>(placedContext_.get());
+    auto piPlacedContext_ =
+        reinterpret_cast<pi_context>(cl::sycl::get_native<cl::sycl::backend::cuda>(placedContext_));
     CUstream streamId = get_stream(queue);
     cublasStatus_t err;
     auto it = handle_helper.cublas_handle_mapper_.find(piPlacedContext_);

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,8 @@ void ContextCallback(void *userData) {`
`97`	`97`	`}`
`98`	`98`
`99`	`99`	`cublasHandle_t CublasScopedContextHandler::get_handle(const cl::sycl::queue &queue) {`
`100`		`- auto piPlacedContext_ = reinterpret_cast<pi_context>(placedContext_.get());`
	`100`	`+ auto piPlacedContext_ =`
	`101`	`+ reinterpret_cast<pi_context>(cl::sycl::get_native<cl::sycl::backend::cuda>(placedContext_));`
`101`	`102`	`CUstream streamId = get_stream(queue);`
`102`	`103`	`cublasStatus_t err;`
`103`	`104`	`auto it = handle_helper.cublas_handle_mapper_.find(piPlacedContext_);`