merged

Author: ksimpson-work

cuda_core/tests/test_compiler.py

@@ -0,0 +1,7 @@
+from cuda.py._compiler import Compiler
+
+def test_compiler_initialization():
+    code = "__device__ int test_func() { return 0; }"
+    compiler = Compiler(code, "c++")
+    assert compiler._handle is not None
+    assert compiler._backend == "nvrtc"

cuda_core/tests/test_device.py

@@ -50,4 +50,4 @@ def test_compute_capability():
     minor = handle_return(cudart.cudaDeviceGetAttribute(
         cudart.cudaDeviceAttr.cudaDevAttrComputeCapabilityMinor, device.device_id))
     expected_cc = ComputeCapability(major, minor)
-    assert device.compute_capability == expected_cc
+    assert device.compute_capability == expected_cc

cuda_core/tests/test_event.py

@@ -35,4 +35,4 @@ def test_is_done():
 def test_handle():
     options = EventOptions()
     event = Event._init(options)
-    assert isinstance(event.handle, int)
+    assert isinstance(event.handle, int)

cuda_core/tests/test_memory.py

@@ -68,4 +68,4 @@ def test_buffer_close():
     buffer = dummy_mr.allocate(size=1024)
     buffer.close()
     assert buffer.handle == 0
-    assert buffer.memory_resource == None
+    assert buffer.memory_resource == None

cuda_core/tests/test_program.py

@@ -47,4 +47,4 @@ def test_program_close():
     code = "extern \"C\" __global__ void my_kernel() {}"
     program = Program(code, "c++")
     program.close()
-    assert program.handle is None
+    assert program.handle is None