microsoft / microxcaling

PyTorch emulation library for Microscaling (MX)-compatible data formats
MIT License
123 stars 14 forks source link

Add the required changes to support mx auto injection. #13

Closed mgolub2-ms closed 6 months ago

mgolub2-ms commented 6 months ago

This renames torch.X and torch.nn.functional.X calls to allow us to replace the calls in the torch module without causing infinite recursion.

This commit also adds a pyproject.toml that enables building a wheel with 'hatch build'