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 1414 from cuda import cuda as driver
1515 from cuda import cudart as runtime
1616 from cuda import nvrtc
17- import traceback
1817
1918
2019class CUDAError (Exception ):
@@ -36,14 +35,7 @@ def _check_error(error, handle=None):
3635 if err == driver .CUresult .CUDA_SUCCESS :
3736 err , desc = driver .cuGetErrorString (error )
3837 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 ()} " )
4739 else :
4840 raise CUDAError (f"unknown error: { error } " )
4941 elif isinstance (error , runtime .cudaError_t ):
You can’t perform that action at this time.
0 commit comments