Oguz Ulgen 79ee6bbde3 Support triton.language.dtype with torch.compile (#121690) ... Putting this PR as an RFC since I have resorted to some horrible hacks in order to make this work. ``` (Pdb) p triton.language.float32 triton.language.fp32 (Pdb) p str(triton.language.float32) 'fp32' (Pdb) p repr(triton.language.float32) 'triton.language.fp32' ``` This means that we need to "rewrite" them for fx graph and inductor execution. This PR allows Mamba2 to work with `torch.compile`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/121690 Approved by: https://github.com/Skylion007		2024-03-12 23:21:46 +00:00
..
_sympy	Skip less replacements (#119570)	2024-02-20 04:10:33 +00:00
backcompat
benchmark	Enable possibly-undefined error code (#118533)	2024-01-30 21:07:01 +00:00
bottleneck
data	Preserve metadata for MutableMapping and MutableSequence in pin_memory and collate_fn (#120553)	2024-03-01 20:43:42 +00:00
hipify	[ROCm] Enable float16/complex32 fft tests on ROCm (#117296)	2024-02-13 22:35:32 +00:00
jit
model_dump
tensorboard	[tensorboard] Handle bfloat16 type in add_histogram (#120087)	2024-03-05 00:27:21 +00:00
viz	[Memory Snapshot] Add Total memory used after allocation in Trace View (#120339)	2024-02-23 16:17:14 +00:00
__init__.py
_config_module.py	Enable local_partial_types (#118467)	2024-01-28 13:38:22 +00:00
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cuda_trace.py
_cxx_pytree.py	[pytree] support X | Y union type in tree_map_only (#120389)	2024-02-22 18:17:13 +00:00
_device.py	Enable local_partial_types (#118467)	2024-01-28 13:38:22 +00:00
_foreach_utils.py	[DTensor] Supported foreach=True for clip_grad_norm_ (#120910)	2024-03-02 00:28:09 +00:00
_freeze.py
_import_utils.py	[CI] Install dill in ci (#116214)	2024-01-24 23:42:35 +00:00
_mode_utils.py
_python_dispatch.py	Clean up mode handling in python dispatcher (#121083)	2024-03-08 00:30:34 +00:00
_pytree.py	[pytree][reland] Require pytree serialized_type_name (#120636)	2024-02-27 06:53:33 +00:00
_stats.py
_traceback.py
_triton.py	Support triton.language.dtype with torch.compile (#121690)	2024-03-12 23:21:46 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py	Enable possibly-undefined error code (#118533)	2024-01-30 21:07:01 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py	Fix venv compatibility issue by updating python_lib_path (#121103)	2024-03-06 17:00:46 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py	fixed flop counter formula for conv transposed backwards pass (#119874)	2024-02-16 02:43:49 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
show_pickle.py
throughput_benchmark.py
weak.py