biondizzle dfd9c10ae9 Fix MHC import: don't import .torch from layers/mhc.py ...

The layers/mhc.py was trying to import kernels.mhc.torch which
failed because our __init__.py was breaking the package. Instead,
just import our mhc_torch_ops which has everything we need.

Also fix __init__.py to explicitly import mhc_pre_torch and
mhc_post_torch from .torch instead of using import *.

2026-05-19 05:36:35 +00:00

..

kernels/linear/nvfp4

Replace autograd.Function with torch.library.custom_op for Dynamo compat

2026-05-19 01:54:48 +00:00

patches

Fix MHC import: don't import .torch from layers/mhc.py

2026-05-19 05:36:35 +00:00

cutedsl_quant_method.py

Replace autograd.Function with torch.library.custom_op for Dynamo compat

2026-05-19 01:54:48 +00:00

nvfp4_cutedsl.py

Replace autograd.Function with torch.library.custom_op for Dynamo compat

2026-05-19 01:54:48 +00:00