Skip to content

GitLab

Commit eebc98fc authored by wenjh's avatar wenjh
Browse files

Merge branch 'develop_v2.10' into release_v2.10

parents d8c4e220 99a1c744
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
transformer_engine/common/gemm/cublaslt_gemm.cu
View file @ eebc98fc
...@@ -1218,6 +1218,8 @@ void nvte_cublas_atomic_gemm(const NVTETensor A, const NVTETensor B, NVTETensor ...@@ -1218,6 +1218,8 @@ void nvte_cublas_atomic_gemm(const NVTETensor A, const NVTETensor B, NVTETensor
} else { // TT } else { // TT
NVTE_ERROR("TT layout not allowed."); NVTE_ERROR("TT layout not allowed.");
} }
const bool use_int8 = is_int8_dtype(inputA->data.dtype) ||
is_int8_dtype(inputB->data.dtype);
const char *NVTE_FORCE_ROCM_GEMM = std::getenv("NVTE_FORCE_ROCM_GEMM"); const char *NVTE_FORCE_ROCM_GEMM = std::getenv("NVTE_FORCE_ROCM_GEMM");
const bool use_fp8 = is_fp8_dtype(inputA->data.dtype) || const bool use_fp8 = is_fp8_dtype(inputA->data.dtype) ||
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment