Fix some issues in Python bindings + skip sass verification in failing tests

Author: shwina

python/cuda_parallel/cuda/parallel/experimental/_bindings.pyx

@@ -1671,13 +1671,17 @@ cdef class DeviceUniqueByKeyBuildResult:
             )
         return storage_sz
 
+    def _get_cubin(self):
+        return self.build_data.cubin[:self.build_data.cubin_size]
+
 # -----------------
 # DeviceRadixSort
 # -----------------
 
 cdef extern from "cccl/c/radix_sort.h":
     cdef struct cccl_device_radix_sort_build_result_t 'cccl_device_radix_sort_build_result_t':
-        pass
+        const char* cubin
+        size_t cubin_size
 
     cdef CUresult cccl_device_radix_sort_build(
         cccl_device_radix_sort_build_result_t *build_ptr,

python/cuda_parallel/cuda/parallel/experimental/algorithms/_merge_sort.py

@@ -6,7 +6,6 @@
 from typing import Callable
 
 import numba
-from numba import types
 
 from .. import _bindings
 from .. import _cccl_interop as cccl

python/cuda_parallel/cuda/parallel/experimental/algorithms/_radix_sort.py

@@ -114,7 +114,13 @@ def __init__(
         self.d_out_values_cccl = cccl.to_cccl_iter(d_out_values_array)
 
         # TODO: decomposer op is not supported for now
-        self.decomposer_op = cccl.to_cccl_op(None, None)
+        self.decomposer_op = cccl.Op(
+            name="",
+            operator_type=cccl.OpKind.STATELESS,
+            ltoir=b"",
+            state_alignment=1,
+            state=None,
+        )
         decomposer_return_type = "".encode("utf-8")
 
         self.build_result = call_build(

python/cuda_parallel/pyproject.toml

@@ -77,3 +77,6 @@ extend = "../../pyproject.toml"
 
 [tool.ruff.lint.isort]
 known-first-party = ["cuda.parallel"]
+
+[tool.pytest.ini_options]
+markers = ["no_verify_sass: skip SASS verification check"]

python/cuda_parallel/tests/conftest.py

@@ -70,11 +70,14 @@ def cuda_stream() -> Stream:
 
 
 @pytest.fixture(scope="function", autouse=True)
-def verify_sass(monkeypatch):
+def verify_sass(request, monkeypatch):
+    if request.node.get_closest_marker("no_verify_sass"):
+        return
+
     import cuda.parallel.experimental._cccl_interop
 
     monkeypatch.setattr(
         cuda.parallel.experimental._cccl_interop,
         "_check_sass",
-        False,  # todo: change to True
+        True,
     )

python/cuda_parallel/tests/test_reduce_api.py

@@ -2,6 +2,8 @@
 #
 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
+import pytest
+
 
 def test_device_reduce():
     # example-begin reduce-min
@@ -215,6 +217,7 @@ def max_g_value(x, y):
     # example-end reduce-struct
 
 
+@pytest.mark.no_verify_sass(reason="LDL/STL instructions emitted for this test.")
 def test_reduce_struct_type_minmax():
     # example-begin reduce-minmax
     import cupy as cp

python/cuda_parallel/tests/test_scan.py

@@ -120,6 +120,7 @@ def op(a, b):
     "force_inclusive",
     [True, False],
 )
+@pytest.mark.no_verify_sass(reason="LDL/STL instructions emitted for this test.")
 def test_scan_struct_type(force_inclusive):
     @gpu_struct
     class XY: