diff --git a/benchmarks/dynamo/Makefile b/benchmarks/dynamo/Makefile index 0ca8a1b792e..c62773280bc 100644 --- a/benchmarks/dynamo/Makefile +++ b/benchmarks/dynamo/Makefile @@ -1,3 +1,9 @@ +# Usage: +# make build-deps TORCHBENCH_MODELS= +# Support install a single torchbench model (e.g., "alexnet"), +# or multiple torchbench model names (e.g., "alexnet basic_gnn_gcn BERT_pytorch"), +# or empty (i.e., "") for installing all torchbench models. + clone-deps: (cd ../../.. \ && (test -e torchvision || git clone --recursive https://github.com/pytorch/vision torchvision) \ @@ -23,11 +29,11 @@ pull-deps: clone-deps build-deps: clone-deps uv pip install astunparse numpy scipy ninja pyyaml mkl mkl-include setuptools cmake \ typing-extensions requests protobuf numba cython scikit-learn librosa - (cd ../../../torchvision && uv pip install -e .) + (cd ../../../torchvision && uv pip install -e . --no-build-isolation) (cd ../../../torchdata && uv pip install -e .) (cd ../../../torchaudio && uv pip install -e . --no-build-isolation) (cd ../../../FBGEMM/fbgemm_gpu && uv pip install -r requirements.txt && uv pip install -e . --no-build-isolation) (cd ../../../torchrec && uv pip install -e .) (cd ../../../detectron2 && uv pip install -e . --no-build-isolation) - (cd ../../../torchbenchmark && python install.py --continue_on_fail) + (cd ../../../torchbenchmark && python install.py --continue_on_fail $(if $(TORCHBENCH_MODELS),models $(TORCHBENCH_MODELS))) uv pip uninstall torchrec-nightly fbgemm-gpu-nightly