tjh.dev
1From 6f92d54e5a544bc34bb07f2808d554a71cc0e4c3 Mon Sep 17 00:00:00 2001
2From: SomeoneSerge <else+aalto@someonex.net>
3Date: Sun, 13 Oct 2024 14:30:19 +0000
4Subject: [PATCH 3/3] nvidia: cudart a systempath
5
6---
7 third_party/nvidia/backend/driver.c | 2 +-
8 third_party/nvidia/backend/driver.py | 5 +++--
9 2 files changed, 4 insertions(+), 3 deletions(-)
10
11diff --git a/third_party/nvidia/backend/driver.c b/third_party/nvidia/backend/driver.c
12index 44524da27..fbdf0d156 100644
13--- a/third_party/nvidia/backend/driver.c
14+++ b/third_party/nvidia/backend/driver.c
15@@ -1,4 +1,4 @@
16-#include "cuda.h"
17+#include <cuda.h>
18 #include <dlfcn.h>
19 #include <stdbool.h>
20 #define PY_SSIZE_T_CLEAN
21diff --git a/third_party/nvidia/backend/driver.py b/third_party/nvidia/backend/driver.py
22index 30fbadb2a..65c0562ed 100644
23--- a/third_party/nvidia/backend/driver.py
24+++ b/third_party/nvidia/backend/driver.py
25@@ -10,7 +10,8 @@ from triton.backends.compiler import GPUTarget
26 from triton.backends.driver import GPUDriver
27
28 dirname = os.path.dirname(os.path.realpath(__file__))
29-include_dir = [os.path.join(dirname, "include")]
30+import shlex
31+include_dir = [*shlex.split("@cudaToolkitIncludeDirs@"), os.path.join(dirname, "include")]
32 libdevice_dir = os.path.join(dirname, "lib")
33 libraries = ['cuda']
34
35@@ -149,7 +150,7 @@ def make_launcher(constants, signature, ids):
36 # generate glue code
37 params = [i for i in signature.keys() if i not in constants]
38 src = f"""
39-#include \"cuda.h\"
40+#include <cuda.h>
41 #include <stdbool.h>
42 #include <Python.h>
43 #include <dlfcn.h>
44--
452.46.0
46