pytorch/torch/utils
Jez Ng 31c0467594 Add Triton CPU as an Inductor backend (#133408)
The goal is to use Inductor-generated kernels to stress test the new Triton CPU backend.

Differential Revision: [D63298968](https://our.internmc.facebook.com/intern/diff/D63298968)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/133408
Approved by: https://github.com/jansel, https://github.com/blaine-rister, https://github.com/malfet
2024-09-26 15:35:26 +00:00
..
_strobelight
_sympy fix mypi in utils/_sympy/functions.py (#136339) 2024-09-20 18:39:16 +00:00
backcompat
benchmark
bottleneck
data
hipify
jit
model_dump
tensorboard Param fixes in docstring (#136097) 2024-09-21 18:56:34 +00:00
viz
__init__.py
_backport_slots.py
_config_module.py
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py
_device.py
_exposed_in.py
_foreach_utils.py
_freeze.py
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_ordered_set.py
_python_dispatch.py [Dynamo] Automatically in-graph traceable tensor subclass ctors (#135151) 2024-09-06 12:23:38 +00:00
_pytree.py
_stats.py
_thunk.py
_traceback.py
_triton.py Add Triton CPU as an Inductor backend (#133408) 2024-09-26 15:35:26 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py Param fixes in docstring (#136097) 2024-09-21 18:56:34 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py Param fixes in docstring (#136097) 2024-09-21 18:56:34 +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 [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00