├── .gitignore ├── INSTALLATION.md ├── README.md ├── configs ├── base_fp16.yaml ├── base_fp32.yaml ├── dynamo_fp16.yaml └── dynamo_fp32.yaml ├── experiments ├── accelerate_script.py ├── base.py ├── base_fp16.py ├── dynamic.py ├── dynamic_fp16.py ├── dynamic_optimized.py ├── dynamic_optimized_fp16.py ├── generate_script.py ├── optimize_forward.py ├── optimize_forward_fp16.py ├── optimize_model.py ├── optimize_model_fp16.py ├── optimize_train_step.py └── optimize_train_step_fp16.py ├── requirements.txt ├── run_experiments.sh ├── scripts ├── cv_classification.py ├── language_modeling.py ├── text_classification.py └── translation.py └── tools ├── summarize.py └── verify_dynamo.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | trained-resnet 3 | -------------------------------------------------------------------------------- /INSTALLATION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/INSTALLATION.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/README.md -------------------------------------------------------------------------------- /configs/base_fp16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/configs/base_fp16.yaml -------------------------------------------------------------------------------- /configs/base_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/configs/base_fp32.yaml -------------------------------------------------------------------------------- /configs/dynamo_fp16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/configs/dynamo_fp16.yaml -------------------------------------------------------------------------------- /configs/dynamo_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/configs/dynamo_fp32.yaml -------------------------------------------------------------------------------- /experiments/accelerate_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/accelerate_script.py -------------------------------------------------------------------------------- /experiments/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/base.py -------------------------------------------------------------------------------- /experiments/base_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/base_fp16.py -------------------------------------------------------------------------------- /experiments/dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/dynamic.py -------------------------------------------------------------------------------- /experiments/dynamic_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/dynamic_fp16.py -------------------------------------------------------------------------------- /experiments/dynamic_optimized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/dynamic_optimized.py -------------------------------------------------------------------------------- /experiments/dynamic_optimized_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/dynamic_optimized_fp16.py -------------------------------------------------------------------------------- /experiments/generate_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/generate_script.py -------------------------------------------------------------------------------- /experiments/optimize_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/optimize_forward.py -------------------------------------------------------------------------------- /experiments/optimize_forward_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/optimize_forward_fp16.py -------------------------------------------------------------------------------- /experiments/optimize_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/optimize_model.py -------------------------------------------------------------------------------- /experiments/optimize_model_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/optimize_model_fp16.py -------------------------------------------------------------------------------- /experiments/optimize_train_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/optimize_train_step.py -------------------------------------------------------------------------------- /experiments/optimize_train_step_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/experiments/optimize_train_step_fp16.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_experiments.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/run_experiments.sh -------------------------------------------------------------------------------- /scripts/cv_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/scripts/cv_classification.py -------------------------------------------------------------------------------- /scripts/language_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/scripts/language_modeling.py -------------------------------------------------------------------------------- /scripts/text_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/scripts/text_classification.py -------------------------------------------------------------------------------- /scripts/translation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/scripts/translation.py -------------------------------------------------------------------------------- /tools/summarize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/tools/summarize.py -------------------------------------------------------------------------------- /tools/verify_dynamo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sgugger/torchdynamo-tests/HEAD/tools/verify_dynamo.py --------------------------------------------------------------------------------