fix hip 6 and add comgr

tinygrad · Jan 25, 2024 · 3b39fa0 · 3b39fa0
1 parent 88eaa5f
commit 3b39fa0
Show file tree

Hide file tree

Showing 2 changed files with 932 additions and 216 deletions.
diff --git a/generate_hip.sh b/generate_hip.sh
@@ -1,5 +1,7 @@
 #!/bin/bash -e
-clang2py /opt/rocm/include/hip/hiprtc.h /opt/rocm/include/hip/hip_runtime_api.h /opt/rocm/include/hip/driver_types.h /opt/rocm/include/hip/hip_ext.h --clang-args="-D__HIP_PLATFORM_AMD__ -I/opt/rocm/include -x c++" -o gpuctypes/hip.py -l /opt/rocm/lib/libhiprtc.so -l /opt/rocm/lib/libamdhip64.so
+clang2py -k acdefstu /opt/rocm/include/amd_comgr/amd_comgr.h /opt/rocm/include/hip/hip_ext.h /opt/rocm/include/hip/hiprtc.h /opt/rocm/include/hip/hip_runtime_api.h /opt/rocm/include/hip/driver_types.h --clang-args="-D__HIP_PLATFORM_AMD__ -I/opt/rocm/include -x c++" -o gpuctypes/hip.py -l /opt/rocm/lib/libhiprtc.so -l /opt/rocm/lib/libamdhip64.so -l /opt/rocm/lib/libamd_comgr.so
+echo "hipDeviceProp_t = hipDeviceProp_tR0600" >> gpuctypes/hip.py
+echo "hipGetDeviceProperties = hipGetDevicePropertiesR0600" >> gpuctypes/hip.py
 grep FIXME_STUB gpuctypes/hip.py || true
 # we can trust HIP is always at /opt/rocm/lib
 #sed -i "s\import ctypes\import ctypes, ctypes.util\g" gpuctypes/hip.py