Skip to content

GitLab

Commit d55fdf7b authored by yuguo's avatar yuguo
Browse files

[DCU] fix compile

parent e45d66a3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
transformer_engine/common/common.cu
View file @ d55fdf7b
......@@ -26,6 +26,7 @@ __global__ void __launch_bounds__(1)
} // namespace
#ifndef __HIP_PLATFORM_AMD__
cudaDataType_t get_cuda_dtype(const transformer_engine::DType t) {
using namespace transformer_engine;
switch (t) {
......@@ -43,6 +44,7 @@ cudaDataType_t get_cuda_dtype(const transformer_engine::DType t) {
NVTE_ERROR("Invalid type");
}
}
#endif
void update_tensor_scale_inv(Tensor *t, cudaStream_t stream) {
if (is_fp8_dtype(t->data.dtype) && is_tensor_scaling(t->scaling_mode)) {
......
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