[release/2.3] Added cublasGemmAlgo_t -> hipblasGemmAlgo_t (#126448)

Added cublasGemmAlgo_t -> hipblasGemmAlgo_t
pytorch · May 23, 2024 · 71dd2de · 71dd2de
1 parent 6cd59f1
commit 71dd2de
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/utils/hipify/cuda_to_hip_mappings.py b/torch/utils/hipify/cuda_to_hip_mappings.py
@@ -454,6 +454,7 @@
         ("cublasDiagType_t", ("hipblasDiagType_t", CONV_TYPE, API_BLAS)),
         ("cublasSideMode_t", ("hipblasSideMode_t", CONV_TYPE, API_BLAS)),
         ("cublasPointerMode_t", ("hipblasPointerMode_t", CONV_TYPE, API_BLAS)),
+        ("cublasGemmAlgo_t", ("hipblasGemmAlgo_t", CONV_TYPE, API_BLAS)),
         (
             "cublasAtomicsMode_t",
             ("hipblasAtomicsMode_t", CONV_TYPE, API_BLAS, HIP_UNSUPPORTED),