[AMD] Reorganize hip-related header files in sgl-kernel (#9320)
This commit is contained in:
@@ -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 = [
|
||||
|
||||
Reference in New Issue
Block a user