NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-29 19:21:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #269 from jayavenkatesh19/main

Browse Source 
remove pynvjitlink references in examples
This commit is contained in:
Oleksandr Pavlyk
2025-09-17 13:54:36 -05:00
committed by GitHub
parent 4642df7006 0f997271f7
commit b88a45f417
2 changed files with 1 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 numba
import numpy as np import numpy as np
from numba import cuda from numba import cuda
from pynvjitlink import patch
class BitsetRing: class BitsetRing:
@@ -90,8 +89,6 @@ def multi_block_bench(state: bench.State):
if __name__ == "__main__": if __name__ == "__main__":
patch.patch_numba_linker(lto=True)
b = bench.register(multi_block_bench) b = bench.register(multi_block_bench)
b.add_int64_axis("ThreadsPerBlock", [64, 128, 192, 256]) b.add_int64_axis("ThreadsPerBlock", [64, 128, 192, 256])
b.add_int64_power_of_two_axis("NumBlocks", [10, 11, 12, 14, 16]) b.add_int64_power_of_two_axis("NumBlocks", [10, 11, 12, 14, 16])

1
python/examples/requirements.txt
View File

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