Fix build error

transformer_engine/common/gemm/rocm_gemm.cu

@@ -954,6 +954,9 @@ void hipblaslt_gemm(const Tensor* inputA, const Tensor* inputB, Tensor* outputD,
   float zero = 0.0;
   float beta = (accumulate) ? one : zero;
 
+  int device_id;
+  NVTE_CHECK_CUDA(hipGetDevice(&device_id));
+
   hipblasLtHandle_t handle = hipBlasLtHandleManager::Instance().GetHandle();
 
   hipblasLtMatmulDesc_t operationDesc = nullptr;