From 62a53f43089922f468531261f197a80a431d2c71 Mon Sep 17 00:00:00 2001 From: Ritwik Das Date: Mon, 3 Oct 2022 01:12:54 -0700 Subject: [PATCH] Compile ROCM kernels with include path --- hatlib/rocm_loader.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/hatlib/rocm_loader.py b/hatlib/rocm_loader.py index 803075f..03a3c70 100644 --- a/hatlib/rocm_loader.py +++ b/hatlib/rocm_loader.py @@ -26,7 +26,9 @@ def compile_rocm_program(rocm_src_path: pathlib.Path, func_name): header_sources=ROCM_HEADER_MAP.values() ) device_properties = hipGetDeviceProperties(0) - hiprtcCompileProgram(prog, [f'--offload-arch={device_properties.gcnArchName}', '-D__HIP_PLATFORM_AMD__']) + hiprtcCompileProgram(prog, [f'--offload-arch={device_properties.gcnArchName}', + '-D__HIP_PLATFORM_AMD__', + '-I/home/ritdas/rocWMMA/library/include/rocwmma/']) code = hiprtcGetCode(prog) return code