Merge pull request #269 from jayavenkatesh19/main

remove pynvjitlink references in examples
This commit is contained in:
Oleksandr Pavlyk
2025-09-17 13:54:36 -05:00
committed by GitHub
2 changed files with 1 additions and 3 deletions

View File

@@ -21,7 +21,6 @@ import cuda.cccl.cooperative.experimental as coop
import numba
import numpy as np
from numba import cuda
from pynvjitlink import patch
class BitsetRing:
@@ -90,8 +89,6 @@ def multi_block_bench(state: bench.State):
if __name__ == "__main__":
patch.patch_numba_linker(lto=True)
b = bench.register(multi_block_bench)
b.add_int64_axis("ThreadsPerBlock", [64, 128, 192, 256])
b.add_int64_power_of_two_axis("NumBlocks", [10, 11, 12, 14, 16])

View File

@@ -5,3 +5,4 @@ nvidia-cutlass
cuda-cccl
cuda-core
cuda-bindings
numba-cuda