Download dependencies/cub/benchmarks/scripts/search.py from camenduru/thrust: direct link, hf CLI and curl.
- Browser
- Download file 319 Bytes
-
https://huggingface.co/camenduru/thrust/resolve/main/dependencies/cub/benchmarks/scripts/search.py
- Command line
-
hf download hf://camenduru/thrust/dependencies/cub/benchmarks/scripts/search.py
-
curl -L -o search.py https://huggingface.co/camenduru/thrust/resolve/main/dependencies/cub/benchmarks/scripts/search.py
319 Bytes
| #!/usr/bin/env python3 | |
| import cub.bench as bench | |
| # TODO: | |
| # - driver version | |
| # - host compiler + version | |
| # - gpu clocks / pm | |
| # - ecc | |
| def main(): | |
| center_estimator = bench.MedianCenterEstimator() | |
| bench.search(bench.BruteForceSeeker(center_estimator, center_estimator)) | |
| if __name__ == "__main__": | |
| main() | |