Yu, Guangye 19a83eacb5 add new API torch.amp.is_autocast_available (#124938) ... # Motivation expose `torch._is_autocast_available` to `torch.amp.is_autocast_available` as a public api. Pull Request resolved: https://github.com/pytorch/pytorch/pull/124938 Approved by: https://github.com/albanD		2024-04-26 08:45:20 +00:00
..
_sympy	[BE]: Optimize min/max/sum comprehensions C419 (#123960)	2024-04-12 23:54:15 +00:00
backcompat
benchmark
bottleneck	[BE]: Optimize min/max/sum comprehensions C419 (#123960)	2024-04-12 23:54:15 +00:00
data	[BE]: TRY002 - Ban raising vanilla exceptions (#124570)	2024-04-21 22:26:40 +00:00
hipify	[ROCm] update hipDataType support and hipify mappings (#120751)	2024-04-23 23:21:56 +00:00
jit	Revert "Fix global flake8 issues (#124771)"	2024-04-26 06:15:17 +00:00
model_dump	[BE]: TRY002 - Ban raising vanilla exceptions (#124570)	2024-04-21 22:26:40 +00:00
tensorboard	[EZ] Update pillow to 10.3.0 (#124614)	2024-04-23 03:22:23 +00:00
viz	[MemoryViz] Improve description of blocks with missing frames (#124784)	2024-04-24 21:16:31 +00:00
__init__.py
_config_module.py
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py	[pytree] add tree_iter function (#123913)	2024-04-16 06:02:08 +00:00
_device.py
_exposed_in.py	Add torch.library.custom_op (#122344)	2024-04-03 18:36:17 +00:00
_foreach_utils.py	add fused_sgd_kernel support for CPU device (#123629)	2024-04-23 08:28:19 +00:00
_freeze.py
_import_utils.py
_mode_utils.py
_python_dispatch.py	Revert "Fix mypy issues in fake_tensor.py (#124428)"	2024-04-26 06:15:17 +00:00
_pytree.py	[pytree] add tree_iter function (#123913)	2024-04-16 06:02:08 +00:00
_stats.py
_traceback.py
_triton.py	[Inductor Intel GPU backend Upstream] Add Inductor Intel GPU backend. (#121895)	2024-04-05 09:05:11 +00:00
_typing_utils.py
_zip.py
backend_registration.py	refactor autocast python APIs (#124479)	2024-04-25 14:33:33 +00:00
bundled_inputs.py	[BE]: TRY002 - Ban raising vanilla exceptions (#124570)	2024-04-21 22:26:40 +00:00
checkpoint.py	add new API torch.amp.is_autocast_available (#124938)	2024-04-26 08:45:20 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py	[CPP Extension] Escape include paths (#122974)	2024-03-30 21:58:29 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py	fix flop counter issue with out parameters (#123768)	2024-04-11 09:39:53 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py	[BE]: Update ruff to 0.4.1 (#124549)	2024-04-21 14:06:23 +00:00
model_zoo.py
show_pickle.py	[BE]: TRY002 - Ban raising vanilla exceptions (#124570)	2024-04-21 22:26:40 +00:00
throughput_benchmark.py
weak.py