Skip to content

GitLab

"git@developer.sourcefind.cn:bw-bestperf/SAM.git" did not exist on "e9f2d5809489d79f5a5c8043c39511ac77df1282"
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) ...@@ -26,6 +26,7 @@ __global__ void __launch_bounds__(1)
} // namespace } // namespace
#ifndef __HIP_PLATFORM_AMD__
cudaDataType_t get_cuda_dtype(const transformer_engine::DType t) { cudaDataType_t get_cuda_dtype(const transformer_engine::DType t) {
using namespace transformer_engine; using namespace transformer_engine;
switch (t) { switch (t) {
...@@ -43,6 +44,7 @@ cudaDataType_t get_cuda_dtype(const transformer_engine::DType t) { ...@@ -43,6 +44,7 @@ cudaDataType_t get_cuda_dtype(const transformer_engine::DType t) {
NVTE_ERROR("Invalid type"); NVTE_ERROR("Invalid type");
} }
} }
#endif
void update_tensor_scale_inv(Tensor *t, cudaStream_t stream) { void update_tensor_scale_inv(Tensor *t, cudaStream_t stream) {
if (is_fp8_dtype(t->data.dtype) && is_tensor_scaling(t->scaling_mode)) { 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