Skip to content

GitLab

Commit bcaa56b9 authored by wenjh's avatar wenjh
Browse files

Fix build error

parent f791516e
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
transformer_engine/common/gemm/rocm_gemm.cu
View file @ bcaa56b9
...@@ -954,6 +954,9 @@ void hipblaslt_gemm(const Tensor* inputA, const Tensor* inputB, Tensor* outputD, ...@@ -954,6 +954,9 @@ void hipblaslt_gemm(const Tensor* inputA, const Tensor* inputB, Tensor* outputD,
float zero = 0.0; float zero = 0.0;
float beta = (accumulate) ? one : zero; float beta = (accumulate) ? one : zero;
int device_id;
NVTE_CHECK_CUDA(hipGetDevice(&device_id));
hipblasLtHandle_t handle = hipBlasLtHandleManager::Instance().GetHandle(); hipblasLtHandle_t handle = hipBlasLtHandleManager::Instance().GetHandle();
hipblasLtMatmulDesc_t operationDesc = nullptr; hipblasLtMatmulDesc_t operationDesc = nullptr;
......
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