Skip to content

Commit 8004f7d

Browse files
authored
Add missing SASS testing changes to cuda.parallel (NVIDIA#4508)
1 parent 8306d99 commit 8004f7d

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

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

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1671,13 +1671,17 @@ cdef class DeviceUniqueByKeyBuildResult:
16711671
)
16721672
return storage_sz
16731673

1674+
def _get_cubin(self):
1675+
return self.build_data.cubin[:self.build_data.cubin_size]
1676+
16741677
# -----------------
16751678
# DeviceRadixSort
16761679
# -----------------
16771680

16781681
cdef extern from "cccl/c/radix_sort.h":
16791682
cdef struct cccl_device_radix_sort_build_result_t 'cccl_device_radix_sort_build_result_t':
1680-
pass
1683+
const char* cubin
1684+
size_t cubin_size
16811685

16821686
cdef CUresult cccl_device_radix_sort_build(
16831687
cccl_device_radix_sort_build_result_t *build_ptr,

0 commit comments

Comments
 (0)