diff --git a/cuburn/ptx.py b/cuburn/ptx.py index 3627bff..b099554 100644 --- a/cuburn/ptx.py +++ b/cuburn/ptx.py @@ -647,7 +647,8 @@ class PTXEntryPoint(PTXFragment): try: self._call(ctx, func, *args, **kwargs) finally: - return ctx.call_teardown(self) + res = ctx.call_teardown(self) + return res class PTXTestFailure(Exception): pass