pytorch

_sympy

Compute bounds for the variables created during codegen (#123100 )

2024-05-08 08:14:06 +00:00

backcompat

[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308 )

2023-02-07 21:10:56 +00:00

benchmark

Adding Compare in torch.utils.benchmark documentation (#125009 )

2024-05-03 00:50:54 +00:00

bottleneck

[BE]: Optimize min/max/sum comprehensions C419 (#123960 )

2024-04-12 23:54:15 +00:00

data

Fix broken docs (#124940 )

2024-04-26 19:24:52 +00:00

hipify

[ROCm] enforce ROCM_VERSION >= 6.0 (#125646 )

2024-05-12 18:01:28 +00:00

jit

Fix global flake8 issues (#124771 )

2024-04-26 15:35:53 +00:00

model_dump

[BE]: TRY002 - Ban raising vanilla exceptions (#124570 )

2024-04-21 22:26:40 +00:00

tensorboard

Remove Caffe2 python (#125143 )

2024-05-10 21:15:43 +00:00

viz

[MemoryViz] Improve description of blocks with missing frames (#124784 )

2024-04-24 21:16:31 +00:00

__init__.py

Swap slots if slots match in swap_tensor (#116128 )

2023-12-21 00:43:30 +00:00

_config_module.py

Enable local_partial_types (#118467 )

2024-01-28 13:38:22 +00:00

_config_typing.pyi

Revert "[dynamo] guarded config (#111299 )" (#115386 )

2023-12-11 19:35:42 +00:00

_content_store.py

Fix recompilation issue with content store (#113533 )

2023-11-13 23:58:13 +00:00

_contextlib.py

Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311 )

2023-11-15 17:40:04 +00:00

_cpp_extension_versioner.py

[BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308 )

2023-02-07 21:10:56 +00:00

_cxx_pytree.py

[pytree] add tree_iter function (#123913 )

2024-04-16 06:02:08 +00:00

_device.py

Enable local_partial_types (#118467 )

2024-01-28 13:38:22 +00:00

_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

[Codemod][python/main_function] caffe2: (#113357 )

2023-11-15 22:17:31 +00:00

_get_clean_triton.py

Add script for removing Inductor dependencies from Inductor generated code (#125811 )

2024-05-10 00:00:25 +00:00

_import_utils.py

[CI] Install dill in ci (#116214 )

2024-01-24 23:42:35 +00:00

_mode_utils.py

Switch most Python RAII guard usages to context manager (#102642 )

2023-06-01 16:28:37 +00:00

_python_dispatch.py

[RELAND] Switch default behavoir of export IR to be predispatch (#125860 )

2024-05-10 17:36:53 +00:00

_pytree.py

[pytree] add tree_iter function (#123913 )

2024-04-16 06:02:08 +00:00

_stats.py

Fast path binary ops in fake tensor (#94047 )

2023-02-07 18:34:24 +00:00

_traceback.py

Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311 )

2023-11-15 17:40:04 +00:00

_triton.py

[Inductor Intel GPU backend Upstream] Add Inductor Intel GPU backend. (#121895 )

2024-04-05 09:05:11 +00:00

_typing_utils.py

Re-enable type checking for distributed_c10d.py (#115223 )

2023-12-09 11:07:54 +00:00

_zip.py

[Codemod][python/main_function] caffe2: (#113357 )

2023-11-15 22:17:31 +00:00

backend_registration.py

add methods for open device in PackedSequence module. (#124923 )

2024-04-26 15:26:20 +00:00

bundled_inputs.py

[BE]: TRY002 - Ban raising vanilla exceptions (#124570 )

2024-04-21 22:26:40 +00:00

checkpoint.py

make torch.amp.autocast more generic (#125103 )

2024-05-08 12:13:26 +00:00

collect_env.py

[BE]: Update flake8 to v6.1.0 and fix lints (#116591 )

2024-01-03 06:04:44 +00:00

cpp_backtrace.py

Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311 )

2023-11-15 17:40:04 +00:00

cpp_extension.py

[ROCm] enforce ROCM_VERSION >= 6.0 (#125646 )

2024-05-12 18:01:28 +00:00

deterministic.py

Add torch.utils.deterministic.fill_uninitialized_memory flag (#111377 )

2023-11-01 16:10:09 +00:00

dlpack.py

[BE]: Update Ruff to 0.0.280 (#105724 )

2023-07-22 23:03:34 +00:00

file_baton.py

Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311 )

2023-11-15 17:40:04 +00:00

flop_counter.py

Add "accurate" FlopCounter implementations for NestedTensor SDPA kernels (#125776 )

2024-05-10 19:49:37 +00:00

hooks.py

[FSDP] mark pre_backward_hook unserializable (#125464 )

2024-05-06 20:20:31 +00:00

mkldnn.py

[BE]: Update flake8 to v6.1.0 and fix lints (#116591 )

2024-01-03 06:04:44 +00:00

mobile_optimizer.py

[BE]: Update ruff to 0.4.1 (#124549 )

2024-04-21 14:06:23 +00:00

model_zoo.py

…

module_tracker.py

More accurate is_bw and prompt parents cleanup for ModuleTracker utils (#125634 )

2024-05-07 20:57:36 +00:00

show_pickle.py

[BE]: TRY002 - Ban raising vanilla exceptions (#124570 )

2024-04-21 22:26:40 +00:00

throughput_benchmark.py

Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311 )

2023-11-15 17:40:04 +00:00

weak.py

[pytorch][PR] introduce WeakHashRef (#115216 )

2023-12-07 17:48:11 +00:00