File tree Expand file tree Collapse file tree 1 file changed +1
-9
lines changed
cuda_core/cuda/core/experimental Expand file tree Collapse file tree 1 file changed +1
-9
lines changed Original file line number Diff line number Diff line change 14
14
from cuda import cuda as driver
15
15
from cuda import cudart as runtime
16
16
from cuda import nvrtc
17
- import traceback
18
17
19
18
20
19
class CUDAError (Exception ):
@@ -36,14 +35,7 @@ def _check_error(error, handle=None):
36
35
if err == driver .CUresult .CUDA_SUCCESS :
37
36
err , desc = driver .cuGetErrorString (error )
38
37
if err == driver .CUresult .CUDA_SUCCESS :
39
- stack = traceback .extract_stack ()
40
- # Get the last 2 frames (excluding the current one)
41
- relevant_stack = stack [- 4 :- 1 ]
42
- stack_info = "\n " .join (
43
- f" File '{ frame .filename } ', line { frame .lineno } , in { frame .name } \n { frame .line } "
44
- for frame in relevant_stack
45
- )
46
- raise CUDAError (f"{ name .decode ()} : { desc .decode ()} \n { stack_info } " )
38
+ raise CUDAError (f"{ name .decode ()} : { desc .decode ()} " )
47
39
else :
48
40
raise CUDAError (f"unknown error: { error } " )
49
41
elif isinstance (error , runtime .cudaError_t ):
You can’t perform that action at this time.
0 commit comments