Searched refs:API_BLAS (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/hipify/ |
| H A D | cuda_to_hip_mappings.py | 3 from .constants import (API_BLAS, API_C10, API_CAFFE2, API_DRIVER, API_FFT,
413 ("cublasHandle_t", ("hipblasHandle_t", CONV_TYPE, API_BLAS)),
414 ("cublasOperation_t", ("hipblasOperation_t", CONV_TYPE, API_BLAS)),
415 ("cublasStatus_t", ("hipblasStatus_t", CONV_TYPE, API_BLAS)),
416 ("cublasFillMode_t", ("hipblasFillMode_t", CONV_TYPE, API_BLAS)),
417 ("cublasDiagType_t", ("hipblasDiagType_t", CONV_TYPE, API_BLAS)),
418 ("cublasSideMode_t", ("hipblasSideMode_t", CONV_TYPE, API_BLAS)),
419 ("cublasPointerMode_t", ("hipblasPointerMode_t", CONV_TYPE, API_BLAS)),
420 ("cublasGemmAlgo_t", ("hipblasGemmAlgo_t", CONV_TYPE, API_BLAS)),
423 ("hipblasAtomicsMode_t", CONV_TYPE, API_BLAS, HIP_UNSUPPORTED),
[all …]
|
| H A D | constants.py | 50 API_BLAS = 39 variable
|