def test_mapped_contextmanager(self): # Check that temporarily mapped memory is unregistered immediately, # such that it can be re-mapped at any time class MappedException(Exception): pass arr = np.zeros(1) ctx = cuda.current_context() ctx.deallocations.clear() with self.check_ignored_exception(ctx): with cuda.mapped(arr) as marr: pass with cuda.mapped(arr) as marr: pass # Should also work inside a `defer_cleanup` block with cuda.defer_cleanup(): with cuda.mapped(arr) as marr: pass with cuda.mapped(arr) as marr: pass # Should also work when breaking out of the block due to an exception try: with cuda.mapped(arr) as marr: raise MappedException except MappedException: with cuda.mapped(arr) as marr: pass
def test_mapped_contextmanager(self): # Check that temporarily mapped memory is unregistered immediately, # such that it can be re-mapped at any time class MappedException(Exception): pass arr = np.zeros(1) ctx = cuda.current_context() ctx.deallocations.clear() with self.check_ignored_exception(ctx): with cuda.mapped(arr) as marr: pass with cuda.mapped(arr) as marr: pass # Should also work inside a `defer_cleanup` block with cuda.defer_cleanup(): with cuda.mapped(arr) as marr: pass with cuda.mapped(arr) as marr: pass # Should also work when breaking out of the block due to an exception try: with cuda.mapped(arr) as marr: raise MappedException except MappedException: with cuda.mapped(arr) as marr: pass