Commit 7d0f5b7f authored by wenjh's avatar wenjh
Browse files

[FP8] Fix build error


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent 9666d263
...@@ -54,6 +54,7 @@ using bf16 = nv_bfloat16; ...@@ -54,6 +54,7 @@ using bf16 = nv_bfloat16;
using fp8e4m3 = __nv_fp8_e4m3; using fp8e4m3 = __nv_fp8_e4m3;
using fp8e5m2 = __nv_fp8_e5m2; using fp8e5m2 = __nv_fp8_e5m2;
using fp8e8m0 = uint8_t; using fp8e8m0 = uint8_t;
using int8 = int8_t;
template <typename T> template <typename T>
struct TypeInfo{ struct TypeInfo{
...@@ -65,7 +66,8 @@ struct TypeInfo{ ...@@ -65,7 +66,8 @@ struct TypeInfo{
bf16, bf16,
fp8e4m3, fp8e4m3,
fp8e5m2, fp8e5m2,
fp8e8m0>; fp8e8m0,
int8>;
template <typename U, DType current> template <typename U, DType current>
struct Helper { struct Helper {
......
...@@ -29,11 +29,6 @@ namespace { ...@@ -29,11 +29,6 @@ namespace {
#include "string_code_util_math_h.h" #include "string_code_util_math_h.h"
#include "string_code_utils_cuh.h" #include "string_code_utils_cuh.h"
#ifdef USE_ROCM
#include "string_code_amd_detail_hip_float8_h.h"
#include "string_code_amd_detail_hip_f8_impl_h.h"
#endif // USE_ROCM
#ifndef USE_ROCM #ifndef USE_ROCM
/*! \brief Latest compute capability that NVRTC supports /*! \brief Latest compute capability that NVRTC supports
* *
......
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