Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
TransformerEngine
Commits
7d0f5b7f
Commit
7d0f5b7f
authored
May 26, 2025
by
wenjh
Browse files
[FP8] Fix build error
Signed-off-by:
wenjh
<
wenjh@sugon.com
>
parent
9666d263
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
6 deletions
+3
-6
tests/cpp/test_common.h
tests/cpp/test_common.h
+3
-1
transformer_engine/common/util/rtc.cpp
transformer_engine/common/util/rtc.cpp
+0
-5
No files found.
tests/cpp/test_common.h
View file @
7d0f5b7f
...
@@ -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
{
...
...
transformer_engine/common/util/rtc.cpp
View file @
7d0f5b7f
...
@@ -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
*
*
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment