rzou 98984422eb [triton_op] fix autotuning (#131363) ...

The problem was we were shoving SymInts into the constant_args side
table. The root problem is that torch.fx.node.base_types, which we use
to determine what can be put in the graph, doesn't actually have SymInt
in it. This PR fixes base_types to include SymInt.

Test Plan:
- tests

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131363
Approved by: https://github.com/oulgen, https://github.com/justinchuby

2024-07-24 14:03:37 +00:00

..

__init__.py

[BE] update type annotations for basic utilities in torch/__init__.py (#129001)

2024-06-24 18:04:38 +00:00

associative_scan.py

[BE] update type annotations for basic utilities in torch/__init__.py (#129001)

2024-06-24 18:04:38 +00:00

auto_functionalize.py

[BE] mypy: disallow untyped decorators (#131428)

2024-07-23 21:50:55 +00:00

cond.py

[Autograd] Cond Higher-Order Operation (#126911)

2024-07-22 23:18:19 +00:00

effects.py

[BE] mypy: disallow untyped decorators (#131428)

2024-07-23 21:50:55 +00:00

flex_attention.py

[BE] mypy: disallow untyped decorators (#131428)

2024-07-23 21:50:55 +00:00

map.py

[Autograd] Cond Higher-Order Operation (#126911)

2024-07-22 23:18:19 +00:00

out_dtype.py

[BE] mypy: disallow untyped decorators (#131428)

2024-07-23 21:50:55 +00:00

strict_mode.py

[BE] update type annotations for basic utilities in torch/__init__.py (#129001)

2024-06-24 18:04:38 +00:00

torchbind.py

[torchbind] add tracing_mode support (#129586)

2024-07-12 18:01:47 +00:00

triton_kernel_wrap.py

[triton_op] fix autotuning (#131363)

2024-07-24 14:03:37 +00:00

utils.py

[Autograd] Cond Higher-Order Operation (#126911)

2024-07-22 23:18:19 +00:00

while_loop.py

[BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199)

2024-07-11 17:30:28 +00:00

wrap.py

[Compile+SAC] Only warn for in-place ops once (#129397)

2024-06-26 07:25:02 +00:00