Joel Schlosser 906fe05895 Naive impls for NJT matmul (#138121) ... Our matmul support is abysmal - let's at least get this working and do it performantly later. Bonus: implements `bmm` as well. jagged <-> padded dense conversions are utilized when possible, and an unbind-based fallback otherwise (the former works with torch.compile and the latter doesn't). Some testing is missing because we don't have factory function support yet :( Pull Request resolved: https://github.com/pytorch/pytorch/pull/138121 Approved by: https://github.com/cpuhrsch		2024-10-17 01:31:46 +00:00
..
_internal	Naive impls for NJT matmul (#138121)	2024-10-17 01:31:46 +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
_comparison.py	Strict shape checking for NJTs with TestCase.assertEqual() (#131898)	2024-07-30 20:05:48 +00:00
_creation.py	Ensure noncontiguous tensor creation tests offsetting (#136396)	2024-10-02 00:40:43 +00:00
_utils.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