Skip to content

GitLab

Commit 99a1c744 authored by wenjh's avatar wenjh
Browse files

Fix undefined use_int8 error 


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent 2bb532fb
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
transformer_engine/common/gemm/cublaslt_gemm.cu
View file @ 99a1c744
......@@ -1218,6 +1218,8 @@ void nvte_cublas_atomic_gemm(const NVTETensor A, const NVTETensor B, NVTETensor
} else { // TT
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 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