File tree Expand file tree Collapse file tree 2 files changed +16
-3
lines changed Expand file tree Collapse file tree 2 files changed +16
-3
lines changed Original file line number Diff line number Diff line change 1
1
# Version management
2
2
3
- # TODO : get rid of release getters
4
-
5
3
"""
6
4
driver_version()
7
5
@@ -68,3 +66,8 @@ function set_runtime_version!()
68
66
@warn """ Due to a bug in Julia (until 1.6.5 and 1.7.1) your environment needs to directly include CUDA_Runtime_jll for this to work."""
69
67
end
70
68
end
69
+
70
+
71
+ # # helpers
72
+
73
+ is_tegra () = Sys. islinux () && isfile (" /etc/nv_tegra_release" )
Original file line number Diff line number Diff line change 24
24
# in addition, CUPTI is not available on older GPUs with recent CUDA toolkits
25
25
function can_use_cupti ()
26
26
sanitize && return false
27
+
27
28
# NVIDIA bug #3964667: CUPTI in CUDA 11.7+ broken for sm_35 devices
28
- capability (device ()) > v " 3.7" || CUDA. runtime_version () < v " 11.7"
29
+ if CUDA. runtime_version () >= v " 11.7" && capability (device ()) <= v " 3.7"
30
+ return false
31
+ end
32
+
33
+ # Tegra requires running as root and and modifying the device tree
34
+ if CUDA. is_tegra ()
35
+ return false
36
+ end
37
+
38
+ return true
29
39
end
30
40
31
41
# precompile the runtime library
You can’t perform that action at this time.
0 commit comments