Dmitry Nikolaev f419067e50 [ROCm] improve sparse addmm, enable complex (#153262) ... PR to: - enable complex data types for sparse matmul on ROCm - fix sparse addmm/baddbmm on ROCm - fix sparse hipification for ROCm - fix/enable sparse tests on ROCm (~40 tests total): ``` test_sparse_csr.py::TestSparseCSRCUDA::test_bmm_cuda_* test_sparse.py::TestSparseCUDA::test_sparse_matmul_cuda_* test_sparse_csr.py::TestSparseCSRCUDA::test_mm_cuda_float64 test_sparse_csr.py::TestSparseCSRCUDA::test_addmm_all_sparse_csr_SparseCS* test_sparse_csr.py::TestSparseCSRCUDA::test_addmm_sizes_all_sparse_csr_* ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/153262 Approved by: https://github.com/jeffdaily, https://github.com/pruthvistony		2025-05-19 22:23:18 +00:00
..
_strobelight
_sympy	[inductor] Fix ModularIndexing assumptions (#152993)	2025-05-08 18:26:45 +00:00
backcompat
benchmark	Clean up conda usage in benchmark scripts (#152552)	2025-04-30 21:27:29 +00:00
bottleneck
data	Refactor torch/utils/data/datapipes/gen_pyi.py with torchgen (#150626)	2025-05-17 06:21:41 +00:00
hipify	[ROCm] improve sparse addmm, enable complex (#153262)	2025-05-19 22:23:18 +00:00
jit
model_dump
serialization
tensorboard	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
viz	[MemoryZ] Show the current and max entries rendered (#153446)	2025-05-14 03:16:12 +00:00
__init__.py
_appending_byte_serializer.py	Check integrity of bytes in AppendingByteSerializer (#152139)	2025-04-26 18:10:58 +00:00
_backport_slots.py
_config_module.py
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_embed_headers.py
_cpp_extension_versioner.py
_cxx_pytree.py	[BE] detect CXX pytree requirement with TorchVersion (#151102)	2025-05-01 18:55:57 +00:00
_device.py
_dtype_abbrs.py	[BE] Migrate dtype_abbrs into one location (#152229)	2025-04-28 03:52:47 +00:00
_exposed_in.py
_filelock.py
_foreach_utils.py
_freeze.py
_functools.py
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_ordered_set.py
_python_dispatch.py
_pytree.py	[BE] detect CXX pytree requirement with TorchVersion (#151102)	2025-05-01 18:55:57 +00:00
_stats.py
_thunk.py
_traceback.py
_triton.py	Revert "has_triton: Use the device interface for detecting Triton availability (#139171)"	2025-05-10 14:46:23 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py	[ca][dynamo] always run eager checkpoint region's recomputation in eager (#153300)	2025-05-16 01:37:48 +00:00
collect_env.py	[BE]: Enable ruff YTT linter for Python version checks (#153547)	2025-05-14 21:09:16 +00:00
cpp_backtrace.py
cpp_extension.py	[ROCm] cpp_extension allow user to override default flags (#152432)	2025-05-15 21:06:18 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py