Fix a CUDA 12.2 test issue

Signed-off-by: Eric Shi <[email protected]>

Author: shi-eric

warp/tests/tile/test_tile_shared_memory.py

@@ -110,7 +110,7 @@ def compute(out: wp.array2d(dtype=float)):
 
     out = wp.empty((DIM_M, DIM_N), dtype=float, device=device)
 
-    wp.load_module(device=device)
+    compute.module.load(device)
 
     wp.capture_begin(device, force_module_load=False)
     wp.launch_tiled(compute, dim=[1], inputs=[out], block_dim=BLOCK_DIM, device=device)