Skip to content

Commit 4ca1b47

Browse files
committed
revert utils
1 parent d00e453 commit 4ca1b47

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

cuda_core/cuda/core/experimental/_utils.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
from cuda import cuda as driver
1515
from cuda import cudart as runtime
1616
from cuda import nvrtc
17-
import traceback
1817

1918

2019
class 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):

0 commit comments

Comments
 (0)