iree
iree copied to clipboard
benchmark_suites/TF/vmfb/minilm-l12-h384-uncased-seqlen128 requires CUDA but builds when disabled
With CUDA disabled, the benchmark_suites/TF/vmfb/minilm-l12-h384-uncased-seqlen128 benchmark is attempting to be compiled anyway. This causes an error like:
iree-compile: Unknown command line argument '--iree-hal-cuda-llvm-target-arch=sm_80'. Try: '/usr/local/google/home/laurenzo/src/iree-build/tools/iree-compile --help'
iree-compile: Did you mean '--iree-llvm-target-abi=sm_80'?
Can be repro'd via:
pip install iree-tools-tf iree-tools-tflite -f https://github.com/iree-org/iree/releases
cmake -DIREE_BUILD_BENCHMARKS=ON .
ninja iree-benchmark-suites -k 500