fixture 'compiler' not found while executing "pytest test_bench.py"

pytorch / benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

BSD 3-Clause "New" or "Revised" License

817 stars 259 forks source link

fixture 'compiler' not found while executing "pytest test_bench.py" #2315

Closed yuxuan-z19 closed 1 week ago

yuxuan-z19 commented 1 week ago

I'm profiling the execution of BERT_pytorch with test_bench.py using PyTorch 2.0.1 with CUDA 11.7, in which the benchmark keeps updated to the latest commit. Scripts listed in README worked perfectly until I tried test_bench.py.

Python 3.10.12, and here are the packages I installed: requirements.txt

Perhaps I missed something. Thanks!

xuzhao9 commented 1 week ago

Hi @yuxuan-z19, can you please test on https://github.com/pytorch/benchmark/pull/2320 ?

yuxuan-z19 commented 1 week ago

Thanks @xuzhao9 , it's working!