Skip to content

Commit f855d39

Browse files
committed
fix hpcc include
Signed-off-by: Ceng23333 <441651826@qq.com>
1 parent 1ea59e9 commit f855d39

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

scripts/metax_env.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,9 +52,16 @@ def set_env_for_metax_gpu(
5252
dirs = [
5353
os.path.join(root, "tools", "cu-bridge", "include"),
5454
os.path.join(root, "include", "hcr"),
55+
# cu-bridge cuComplex.h includes "hcComplex.h" from HPCC include/common
56+
os.path.join(root, "include", "common"),
57+
# cu-bridge cusparse wrapper includes "hcsparse.h" under include/hcsparse
58+
os.path.join(root, "include", "hcsparse"),
59+
# cu-bridge cublasLt wrapper includes "hcblasLt.h" under include/hcblas
60+
os.path.join(root, "include", "hcblas"),
61+
# cu-bridge cusolver wrapper includes "hcsolver_common.h" under include/hcsolver
62+
os.path.join(root, "include", "hcsolver"),
5563
os.path.join(root, "include"),
5664
]
5765
for var in ("CPATH", "CPLUS_INCLUDE_PATH", "C_INCLUDE_PATH"):
5866
_prepend_path_var(var, dirs)
5967
return
60-

0 commit comments

Comments
 (0)