Skip to content

Commit b88a45f

Browse files
Merge pull request #269 from jayavenkatesh19/main
remove pynvjitlink references in examples
2 parents 4642df7 + 0f99727 commit b88a45f

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

python/examples/cccl_cooperative_block_reduce.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
import numba
2222
import numpy as np
2323
from numba import cuda
24-
from pynvjitlink import patch
2524

2625

2726
class BitsetRing:
@@ -90,8 +89,6 @@ def launcher(launch: bench.Launch):
9089

9190

9291
if __name__ == "__main__":
93-
patch.patch_numba_linker(lto=True)
94-
9592
b = bench.register(multi_block_bench)
9693
b.add_int64_axis("ThreadsPerBlock", [64, 128, 192, 256])
9794
b.add_int64_power_of_two_axis("NumBlocks", [10, 11, 12, 14, 16])

python/examples/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,4 @@ nvidia-cutlass
55
cuda-cccl
66
cuda-core
77
cuda-bindings
8+
numba-cuda

0 commit comments

Comments
 (0)