add cuda support

tinygrad · Nov 25, 2023 · 971a74e · 971a74e
1 parent 7f81684
commit 971a74e
Show file tree

Hide file tree

Showing 4 changed files with 4,858 additions and 31 deletions.
diff --git a/README b/README
@@ -0,0 +1,4 @@
+Current version is:
+
+* ROCm 5.7.1
+* cuda 11.5
diff --git a/generate.sh b/generate.sh
@@ -1,2 +1,4 @@
 #!/bin/bash
 clang2py /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" -o gpuctypes/hip.py -l /opt/rocm/lib/libhiprtc.so -l /opt/rocm/lib/libamdhip64.so
+clang2py /usr/include/cuda.h /usr/include/nvrtc.h -o gpuctypes/cuda.py -l /usr/lib/x86_64-linux-gnu/libcuda.so -l /usr/lib/x86_64-linux-gnu/libnvrtc.so
+grep FIXME_STUB gpuctypes/*