TensorRT-LLMs/tests/unittest/trt/quantization
Barry Kang 26793e3569
[https://nvbugs/5289907][fix] Restore per-channel pre-quant (#4545)
* Restore per-channel pre-quant

Signed-off-by: Barry Kang <43644113+Barry-Delaney@users.noreply.github.com>

* Update TRT test script

Signed-off-by: Barry Kang <43644113+Barry-Delaney@users.noreply.github.com>

* Fix pre-commit

Signed-off-by: Barry Kang <43644113+Barry-Delaney@users.noreply.github.com>

---------

Signed-off-by: Barry Kang <43644113+Barry-Delaney@users.noreply.github.com>
2025-05-23 19:46:53 +08:00
..
__init__.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
_utils.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_fp8_quantization.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_fp8_rowwise_gemm.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_functional.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_mode.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_moe_weight_only_groupwise_quant_matmul.py [https://nvbugs/5289907][fix] Restore per-channel pre-quant (#4545) 2025-05-23 19:46:53 +08:00
test_qserve_gemm.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_quant_layer.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_quant.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_smooth_quant_gemm.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_smooth_quant_layer_norm.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_smooth_quant_rms_norm.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_weight_only_groupwise_quant_matmul.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_weight_only_quant_matmul.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00