diff --git a/test/quantization/test_mixed_precision.py b/test/prototype/test_mixed_precision.py similarity index 93% rename from test/quantization/test_mixed_precision.py rename to test/prototype/test_mixed_precision.py index 8afd022d3c..ce575722ee 100644 --- a/test/quantization/test_mixed_precision.py +++ b/test/prototype/test_mixed_precision.py @@ -4,7 +4,7 @@ import torch.nn as nn from torchao.quantization import quantize_, int8_weight_only, int4_weight_only from torchao.quantization.utils import compute_error -from torchao.quantization.prototype.mixed_precision.scripts.naive_intNwo import intN_weight_only +from torchao.prototype.quantization.mixed_precision.scripts import intN_weight_only _CUDA_IS_AVAILABLE = torch.cuda.is_available() diff --git a/torchao/quantization/prototype/mixed_precision/__init__.py b/torchao/prototype/quantization/__init__.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/__init__.py rename to torchao/prototype/quantization/__init__.py diff --git a/torchao/quantization/prototype/mixed_precision/README.md b/torchao/prototype/quantization/mixed_precision/README.md similarity index 100% rename from torchao/quantization/prototype/mixed_precision/README.md rename to torchao/prototype/quantization/mixed_precision/README.md diff --git a/torchao/prototype/quantization/mixed_precision/__init__.py b/torchao/prototype/quantization/mixed_precision/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/torchao/quantization/prototype/mixed_precision/scripts/BO_acc_modelsize.py b/torchao/prototype/quantization/mixed_precision/scripts/BO_acc_modelsize.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/BO_acc_modelsize.py rename to torchao/prototype/quantization/mixed_precision/scripts/BO_acc_modelsize.py diff --git a/torchao/quantization/prototype/mixed_precision/scripts/BO_acc_throughput.py b/torchao/prototype/quantization/mixed_precision/scripts/BO_acc_throughput.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/BO_acc_throughput.py rename to torchao/prototype/quantization/mixed_precision/scripts/BO_acc_throughput.py diff --git a/torchao/quantization/prototype/mixed_precision/scripts/Llama3-8B_initial_samples.json b/torchao/prototype/quantization/mixed_precision/scripts/Llama3-8B_initial_samples.json similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/Llama3-8B_initial_samples.json rename to torchao/prototype/quantization/mixed_precision/scripts/Llama3-8B_initial_samples.json diff --git a/torchao/quantization/prototype/mixed_precision/scripts/Llama3-8B_parameters.json b/torchao/prototype/quantization/mixed_precision/scripts/Llama3-8B_parameters.json similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/Llama3-8B_parameters.json rename to torchao/prototype/quantization/mixed_precision/scripts/Llama3-8B_parameters.json diff --git a/torchao/quantization/prototype/mixed_precision/scripts/Mistral-7B_initial_samples.json b/torchao/prototype/quantization/mixed_precision/scripts/Mistral-7B_initial_samples.json similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/Mistral-7B_initial_samples.json rename to torchao/prototype/quantization/mixed_precision/scripts/Mistral-7B_initial_samples.json diff --git a/torchao/quantization/prototype/mixed_precision/scripts/Mistral-7B_parameters.json b/torchao/prototype/quantization/mixed_precision/scripts/Mistral-7B_parameters.json similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/Mistral-7B_parameters.json rename to torchao/prototype/quantization/mixed_precision/scripts/Mistral-7B_parameters.json diff --git a/torchao/quantization/prototype/mixed_precision/scripts/__init__.py b/torchao/prototype/quantization/mixed_precision/scripts/__init__.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/__init__.py rename to torchao/prototype/quantization/mixed_precision/scripts/__init__.py diff --git a/torchao/quantization/prototype/mixed_precision/scripts/fit.py b/torchao/prototype/quantization/mixed_precision/scripts/fit.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/fit.py rename to torchao/prototype/quantization/mixed_precision/scripts/fit.py diff --git a/torchao/quantization/prototype/mixed_precision/scripts/hessian_grad.py b/torchao/prototype/quantization/mixed_precision/scripts/hessian_grad.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/hessian_grad.py rename to torchao/prototype/quantization/mixed_precision/scripts/hessian_grad.py diff --git a/torchao/quantization/prototype/mixed_precision/scripts/hessian_vhp.py b/torchao/prototype/quantization/mixed_precision/scripts/hessian_vhp.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/hessian_vhp.py rename to torchao/prototype/quantization/mixed_precision/scripts/hessian_vhp.py diff --git a/torchao/quantization/prototype/mixed_precision/scripts/mp_quant_eval.py b/torchao/prototype/quantization/mixed_precision/scripts/mp_quant_eval.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/mp_quant_eval.py rename to torchao/prototype/quantization/mixed_precision/scripts/mp_quant_eval.py diff --git a/torchao/quantization/prototype/mixed_precision/scripts/naive_intNwo.py b/torchao/prototype/quantization/mixed_precision/scripts/naive_intNwo.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/naive_intNwo.py rename to torchao/prototype/quantization/mixed_precision/scripts/naive_intNwo.py diff --git a/torchao/quantization/prototype/mixed_precision/scripts/utils.py b/torchao/prototype/quantization/mixed_precision/scripts/utils.py similarity index 100% rename from torchao/quantization/prototype/mixed_precision/scripts/utils.py rename to torchao/prototype/quantization/mixed_precision/scripts/utils.py