Skip to content

Commit 1c57644

Browse files
authored
[rocm7.1_internal_testing] Define datatypes when ROCM_VERSION >= 70000 (#2639)
Fixes SWDEV-543698 Cherry-picked from #2468 This PR fixes the errors like below: [rank7]: RuntimeError: /tmp/comgr-c3c81b/input/CompileSourceejOPx6:34:8: error: unknown type name 'uint64_t'; did you mean '__hip_internal::uint64_t'? [rank7]: 34 | if(((uint64_t) t0.data) % (4 * sizeof(half)) != 0) flag_vec4 = false; Earlier uint64_t was defined in HIP headers in std namespace. Now it is moved to __hip_internal namespace in hip headers. This change is made in ROCm 7.0. Fixes https://ontrack-internal.amd.com/browse/SWDEV-543698
1 parent 681e60e commit 1c57644

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/csrc/jit/codegen/fuser/cuda/resource_strings.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ cases*/
1212

1313
static constexpr auto bfloat16_type_string = "__nv_bfloat16";
1414

15-
#if defined(USE_ROCM)
15+
#if defined(USE_ROCM) && ROCM_VERSION < 70000
1616
static auto type_declarations_template = at::jit::CodeTemplate(R"(
1717
${HalfHeader}
1818
${BFloat16Header}

0 commit comments

Comments
 (0)