pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Pearu Peterson 345578afb4 Add int8 support to bsr_dense_addmm and bsr_dense_mm Triton kernels (#133855 )

As in the title. In addition, the PR introduces `_int_bsr_dense_addmm` that is equivalent to `bsr_dense_addmm` except for int8 inputs the operation result is int32 tensor (similar to existing `_int_mm`).

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133855
Approved by: https://github.com/cpuhrsch

2024-08-21 20:44:40 +00:00

__init__.py

[BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771 )

2024-08-01 17:07:14 +00:00

_semi_structured_conversions.py

Enable UFMT on all of torch/sparse (#130545 )

2024-07-15 22:35:52 +00:00

_semi_structured_ops.py

[BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771 )

2024-08-01 17:07:14 +00:00

_triton_ops_meta.py

Add int8 support to bsr_dense_addmm and bsr_dense_mm Triton kernels (#133855 )