Searched refs:blasPreferredBackend (Results 1 – 4 of 4) sorted by relevance
595 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in bgemm_internal()611 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in bgemm_internal()622 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in bgemm_internal()638 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in bgemm_internal()654 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in bgemm_internal()665 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in bgemm_internal()995 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in gemm_internal()1011 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in gemm_internal()1022 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in gemm_internal()1038 if (at::globalContext().blasPreferredBackend() == BlasBackend::Cublaslt) { in gemm_internal()[all …]
266 at::BlasBackend Context::blasPreferredBackend() { in blasPreferredBackend() function in at::Context
220 at::BlasBackend blasPreferredBackend();
2073 return at::globalContext().blasPreferredBackend(); in initModule()