pytorch/torch/cuda
Nichols A. Romero dcc502f376 [ROCm][TunableOp] Add bias data type to params signature. (#146227)
Add bias vector data type in TunableOp params signature.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/146227
Approved by: https://github.com/jeffdaily
2025-03-11 18:31:22 +00:00
..
amp
__init__.py Initial implementation of host memory stats (#147660) 2025-03-05 16:13:19 +00:00
_gpu_trace.py [BE][Easy] enable PYFMT for torch/[a-s]*/ (#138447) 2024-12-23 14:04:00 +00:00
_memory_viz.py [BE][Easy] enable PYFMT for torch/[a-s]*/ (#138447) 2024-12-23 14:04:00 +00:00
_sanitizer.py PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202) 2025-01-20 22:37:26 +00:00
_utils.py
comm.py
error.py
gds.py [BE] Upgrade to mypy 1.14 (#145966) 2025-03-04 20:58:26 +00:00
graphs.py Revert "Implement cuda graphs implementation of torch.cond and torch.while_loop (#140979)" 2025-02-13 18:04:26 +00:00
jiterator.py PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202) 2025-01-20 22:37:26 +00:00
memory.py Initial implementation of host memory stats (#147660) 2025-03-05 16:13:19 +00:00
nccl.py PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202) 2025-01-20 22:37:26 +00:00
nvtx.py Inductor annotations (#130429) 2024-12-10 08:53:39 +00:00
profiler.py
random.py PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202) 2025-01-20 22:37:26 +00:00
sparse.py
streams.py Support with statement on torch.Stream (#140138) 2025-01-10 02:05:19 +00:00
tunable.py [ROCm][TunableOp] Add bias data type to params signature. (#146227) 2025-03-11 18:31:22 +00:00