tests/quantization/test_torch_compile_utils.py · 48eae6f4204dbdca26e6c1f0c8dc64caa0e48f08 · renzhc / diffusers_dcu

[tests] tests for compilation + quantization (bnb) (#11672) · b6f79330

Sayak Paul authored Jun 11, 2025



* start adding compilation tests for quantization.

* fixes

* make common utility.

* modularize.

* add group offloading+compile

* xfail

* update

* Update tests/quantization/test_torch_compile_utils.py
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

* fixes

---------
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

b6f79330

test_torch_compile_utils.py 3.28 KB

Replace test_torch_compile_utils.py