| .. |
|
_sympy
|
Compute bounds for the variables created during codegen (#123100)
|
2024-05-08 08:14:06 +00:00 |
|
backcompat
|
|
|
|
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
|
[BE]: Update ruff to v0.4.4 (#125031)
|
2024-05-12 20:02:37 +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
|
|
|
|
_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
|
|
|
|
_mode_utils.py
|
|
|
|
_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
|
|
|
|
_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
|
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
|
|
|
|
cpp_backtrace.py
|
|
|
|
cpp_extension.py
|
use shutil.which in _find_cuda_home (#126060)
|
2024-05-13 17:38:17 +00:00 |
|
deterministic.py
|
|
|
|
dlpack.py
|
|
|
|
file_baton.py
|
|
|
|
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
|
|
|
|
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
|
|
|
|
weak.py
|
|
|
3183d65ac0