pytorch · jainapurva · Oct 29, 2024 · Oct 28, 2024 · Oct 29, 2024
diff --git a/test/quantization/test_mixed_precision.py → test/prototype/test_mixed_precision.py b/test/quantization/test_mixed_precision.py → 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/...ion/prototype/mixed_precision/__init__.py → torchao/prototype/quantization/__init__.py b/...ion/prototype/mixed_precision/__init__.py → torchao/prototype/quantization/__init__.py
diff --git a/...ation/prototype/mixed_precision/README.md → ...pe/quantization/mixed_precision/README.md b/...ation/prototype/mixed_precision/README.md → ...pe/quantization/mixed_precision/README.md
diff --git a/torchao/prototype/quantization/mixed_precision/__init__.py b/torchao/prototype/quantization/mixed_precision/__init__.py
diff --git a/...xed_precision/scripts/BO_acc_modelsize.py → ...xed_precision/scripts/BO_acc_modelsize.py b/...xed_precision/scripts/BO_acc_modelsize.py → ...xed_precision/scripts/BO_acc_modelsize.py
diff --git a/...ed_precision/scripts/BO_acc_throughput.py → ...ed_precision/scripts/BO_acc_throughput.py b/...ed_precision/scripts/BO_acc_throughput.py → ...ed_precision/scripts/BO_acc_throughput.py
diff --git a/...on/scripts/Llama3-8B_initial_samples.json → ...on/scripts/Llama3-8B_initial_samples.json b/...on/scripts/Llama3-8B_initial_samples.json → ...on/scripts/Llama3-8B_initial_samples.json
diff --git a/...ecision/scripts/Llama3-8B_parameters.json → ...ecision/scripts/Llama3-8B_parameters.json b/...ecision/scripts/Llama3-8B_parameters.json → ...ecision/scripts/Llama3-8B_parameters.json
diff --git a/...n/scripts/Mistral-7B_initial_samples.json → ...n/scripts/Mistral-7B_initial_samples.json b/...n/scripts/Mistral-7B_initial_samples.json → ...n/scripts/Mistral-7B_initial_samples.json
diff --git a/...cision/scripts/Mistral-7B_parameters.json → ...cision/scripts/Mistral-7B_parameters.json b/...cision/scripts/Mistral-7B_parameters.json → ...cision/scripts/Mistral-7B_parameters.json
diff --git a/...otype/mixed_precision/scripts/__init__.py → ...ation/mixed_precision/scripts/__init__.py b/...otype/mixed_precision/scripts/__init__.py → ...ation/mixed_precision/scripts/__init__.py
diff --git a/.../prototype/mixed_precision/scripts/fit.py → ...antization/mixed_precision/scripts/fit.py b/.../prototype/mixed_precision/scripts/fit.py → ...antization/mixed_precision/scripts/fit.py
diff --git a/...e/mixed_precision/scripts/hessian_grad.py → ...n/mixed_precision/scripts/hessian_grad.py b/...e/mixed_precision/scripts/hessian_grad.py → ...n/mixed_precision/scripts/hessian_grad.py
diff --git a/...pe/mixed_precision/scripts/hessian_vhp.py → ...on/mixed_precision/scripts/hessian_vhp.py b/...pe/mixed_precision/scripts/hessian_vhp.py → ...on/mixed_precision/scripts/hessian_vhp.py
diff --git a/.../mixed_precision/scripts/mp_quant_eval.py → .../mixed_precision/scripts/mp_quant_eval.py b/.../mixed_precision/scripts/mp_quant_eval.py → .../mixed_precision/scripts/mp_quant_eval.py
diff --git a/...e/mixed_precision/scripts/naive_intNwo.py → ...n/mixed_precision/scripts/naive_intNwo.py b/...e/mixed_precision/scripts/naive_intNwo.py → ...n/mixed_precision/scripts/naive_intNwo.py
diff --git a/...rototype/mixed_precision/scripts/utils.py → ...tization/mixed_precision/scripts/utils.py b/...rototype/mixed_precision/scripts/utils.py → ...tization/mixed_precision/scripts/utils.py