[AMD] Reorganize hip-related header files in sgl-kernel (#9320)

This commit is contained in:
Hubert Lu
2025-08-18 16:53:44 -07:00
committed by GitHub
parent c2fbf60f39
commit c6c379ab31
14 changed files with 18 additions and 17 deletions

View File

@@ -72,6 +72,9 @@ if amdgpu_target not in ["gfx942", "gfx950"]:
)
sys.exit(1)
fp8_macro = (
"-DHIP_FP8_TYPE_FNUZ" if amdgpu_target == "gfx942" else "-DHIP_FP8_TYPE_E4M3"
)
hipcc_flags = [
"-DNDEBUG",
@@ -80,10 +83,10 @@ hipcc_flags = [
"-Xcompiler",
"-fPIC",
"-std=c++17",
"-D__HIP_PLATFORM_AMD__=1",
f"--amdgpu-target={amdgpu_target}",
"-DENABLE_BF16",
"-DENABLE_FP8",
fp8_macro,
]
ext_modules = [