NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove pynvjitlink references in examples

Browse Source 
Signed-off-by: Jaya Venkatesh <jjayabaskar@nvidia.com>
This commit is contained in:
Jaya Venkatesh
2025-09-08 16:00:19 -07:00
parent 4642df7006
commit bfa6a6c7c6
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/examples/cccl_cooperative_block_reduce.py
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])