xfail tests if CUDA not available

Author: ianthomas23

rtxpy/rtx.py

@@ -39,14 +39,15 @@ def __init__(self):
             c_lib.traceRTX.restype = ctypes.c_int
             c_lib.cleanRTX.restype = ctypes.c_int
             c_lib.getHashRTX.restype = ctypes.c_uint64
-            if c_lib.initRTX():
-                free_optix_resources()
-                raise RuntimeError("Failed to initialize RTX library")
-            else:
-                atexit.register(free_optix_resources)
         except:
             raise RuntimeError("Failed to load RTX library")
 
+        if c_lib.initRTX():
+            free_optix_resources()
+            raise RuntimeError("Failed to initialize RTX library")
+        else:
+            atexit.register(free_optix_resources)
+
     def build(self, hashValue, vertexBuffer, indexBuffer):
         if has_cupy and isinstance(vertexBuffer, cupy.ndarray):
             vb = ctypes.c_ulonglong(vertexBuffer.data.ptr)

rtxpy/tests/test_simple.py

@@ -21,7 +21,13 @@ def test_simple(test_cupy):
     rays = backend.float32([0.33,0.33,100,0,0,0,-1,1000])
     hits =backend.float32([0,0,0,0])
 
-    optix = RTX()
+    try:
+        optix = RTX()
+    except RuntimeError as e:
+        # RTX fails to initialize if CUDA not available.
+        if str(e) == "Failed to initialize RTX library":
+            pytest.xfail("CUDA not available")
+        raise
 
     res = optix.build(0, verts, triangles)
     assert res == 0