5biwan

If I need torch version, Do I must -reiplemented torch version?

**Describe the bug** The paths for the custom_bwd and custom_fwd methods have changed, and their arguments have been updated. Before ``` from torch.cuda.amp import custom_bwd, custom_fwd ``` After (need to...

Hello, In auto_gtpq_nn_modules_triton_utils_kernels.py, it is suggested to modify the usage of custom_bwd and custom_fwd from torch.cuda.amp as they have been moved under torch.amp and their arguments have changed. If further...