Pearu Peterson 8f2a4cc4b1 Tune bsr_dense_addmm for int8 inputs on A100 (#136088) ... As in the title. The tuning is done for dimensions 1280 and 5120 that are used in Vit-H. Pull Request resolved: https://github.com/pytorch/pytorch/pull/136088 Approved by: https://github.com/cpuhrsch		2024-09-25 00:24:12 +00:00
..
__init__.py	SparseCsrCUDA: cuDSS backend for linalg.solve (#129856)	2024-08-22 07:57:30 +00:00
_semi_structured_conversions.py
_semi_structured_ops.py
_triton_ops.py	Tune bsr_dense_addmm for int8 inputs on A100 (#136088)	2024-09-25 00:24:12 +00:00
_triton_ops_meta.py	Tune bsr_dense_addmm for int8 inputs on A100 (#136088)	2024-09-25 00:24:12 +00:00
semi_structured.py	[BE]: Update mypy to 1.11.2 (#133816)	2024-09-16 19:44:11 +00:00