..
_awaits
_C
[Profiler] add missing field device_resource_id ( #121480 )
2024-03-12 21:42:53 +00:00
_C_flatbuffer
_custom_op
_decomp
Revert "Batch Norm Consolidation ( #116092 )"
2024-03-11 22:22:41 +00:00
_dispatch
_dynamo
Revert "[dynamo][guards] Use lazy variable tracker for func defaults ( #121388 )"
2024-03-12 21:31:18 +00:00
_export
[pt2 export]fix name collision on constant name ( #121145 )
2024-03-11 20:40:59 +00:00
_functorch
[dynamo] Tweak naming for module hook bw_state ( #121609 )
2024-03-12 16:27:56 +00:00
_higher_order_ops
Support tt.reduce in Triton kernel analysis pass ( #121706 )
2024-03-12 11:38:28 +00:00
_inductor
[easy] Refactor MultiOutput. codegen_list_tuple_access to use subclass type checks ( #121662 )
2024-03-12 19:27:56 +00:00
_lazy
_library
Fix FallbackKernel behavior on mutable ops ( #118649 )
2024-02-09 19:01:54 +00:00
_logging
Switch TORCH_TRACE to accept a directory by default ( #121331 )
2024-03-06 22:46:18 +00:00
_numpy
Fix dynamo failure w/ astype ( #117952 )
2024-02-03 08:10:15 +00:00
_prims
add decomposition for frexp ( #119217 )
2024-02-23 21:52:42 +00:00
_prims_common
Handle transposition pattern seen in SDPA with unbacked SymInts ( #121005 )
2024-03-01 18:58:19 +00:00
_refs
[dynamo bug burndown] update tensor creation to support sequences of tensors ( #120872 )
2024-03-02 02:22:59 +00:00
_subclasses
[export] improve binary op fast path broadcast check ( #121546 )
2024-03-09 01:49:42 +00:00
_vendor
amp
Remove device assert in Gradscaler ( #119362 )
2024-02-22 08:02:18 +00:00
ao
[pytorch] optional zero points on dequantize per channel ( #121724 )
2024-03-12 19:54:11 +00:00
autograd
[Profiler] add missing field device_resource_id ( #121480 )
2024-03-12 21:42:53 +00:00
backends
[TorchElastic] Refactoring to support non-default logging strategy ( #120691 )
2024-02-29 20:59:17 +00:00
compiler
[torch.export] Support is_compiling() flag for non-strict mode ( #119602 )
2024-02-29 05:52:51 +00:00
contrib
cpu
csrc
[dynamo][guards-cpp-refactor] Directly call root guard manager in eval_frame ( #121622 )
2024-03-12 17:09:11 +00:00
cuda
[CUDA graphs] Pool argument for make_graphed_callables ( #121475 )
2024-03-09 00:15:38 +00:00
distributed
[dtensor] move early return check into redistribute autograd function ( #121653 )
2024-03-12 17:37:30 +00:00
distributions
Bugfix to MixtureSameFamily's _pad_mixture_dimension ( #118947 )
2024-02-06 16:24:22 +00:00
export
Sanity check for non-strict ( #121687 )
2024-03-12 18:21:32 +00:00
fft
func
Let torch dynamo inline torch.func.grad ( #118407 )
2024-02-28 20:05:00 +00:00
futures
fx
Simplify guards using info from previous guards ( #121463 )
2024-03-12 04:22:20 +00:00
jit
Revert "Batch Norm Consolidation ( #116092 )"
2024-03-11 22:22:41 +00:00
legacy
lib
Remove unneeded linking of torch_shm_manager in CMake ( #119540 )
2024-02-11 06:33:35 +00:00
linalg
Add links to _ex variants in all linalg functions that support them ( #121451 )
2024-03-08 12:19:16 +00:00
masked
monitor
mps
multiprocessing
nested
[NJT] support chunk on batch dim ( #119713 )
2024-03-05 17:57:50 +00:00
nn
Add complex support to parametrizations.spectral_norm ( #121452 )
2024-03-08 19:17:20 +00:00
onnx
Add support to save safetensors checkpoint directly into onnx ( #121001 )
2024-03-11 15:21:59 +00:00
optim
Add ASGD capturable API for forloop ( #121264 )
2024-03-08 00:00:30 +00:00
package
Release the GIL in serialization when it is safe to do so ( #120818 )
2024-03-01 22:37:26 +00:00
profiler
[profiler] Fix recorded profiler step number ( #121127 )
2024-03-09 06:54:51 +00:00
quantization
signal
Clarifying windows cosine behaviour in the documentation ( #119444 )
2024-02-09 05:57:44 +00:00
sparse
[sparse] semi-structured sparse refactor ( #117302 )
2024-02-14 01:10:40 +00:00
special
testing
[DTensor] Moved Transformer sharding to staticmethod ( #121660 )
2024-03-12 15:08:57 +00:00
utils
Clean up mode handling in python dispatcher ( #121083 )
2024-03-08 00:30:34 +00:00
xpu
[DeviceIndex][7/N] Use DeviceIndex in XPU ( #120576 )
2024-02-29 05:54:23 +00:00
__config__.py
__future__.py
Update nn.Module._apply to not gate on should_use_set_data when swap_tensors is set ( #120659 )
2024-02-28 00:59:34 +00:00
__init__.py
Changed cublas repdocubility URL ( #121534 )
2024-03-08 23:46:21 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py
_deploy.py
[Lint] replace [assigment] with [method-assign] for methods ( #119706 )
2024-02-13 02:06:04 +00:00
_guards.py
[dynamo] Fix support for nn.Parameter constructor (part 1) ( #120163 )
2024-03-11 05:14:42 +00:00
_jit_internal.py
[jit][perf] Reduce lookupInModule overhead. ( #119145 )
2024-02-05 18:01:00 +00:00
_linalg_utils.py
_lobpcg.py
[Lint] replace [assigment] with [method-assign] for methods ( #119706 )
2024-02-13 02:06:04 +00:00
_lowrank.py
_meta_registrations.py
add int8 packed gemm support on CPU device ( #118056 )
2024-03-07 08:41:43 +00:00
_namedtensor_internals.py
_ops.py
Don't cache predispatch kernels ( #121712 )
2024-03-12 18:05:59 +00:00
_python_dispatcher.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py
update the tensor.scatter_ doc ( #120169 )
2024-02-23 02:51:55 +00:00
_tensor_str.py
Add sparse compressed meta tensor support ( #120707 )
2024-03-01 13:28:47 +00:00
_tensor.py
Add assign argument to torch.Tensor.module_load ( #121158 )
2024-03-06 01:32:06 +00:00
_torch_docs.py
Fix the default value of side in torch.searchsorted ( #120066 )
2024-02-22 19:35:17 +00:00
_utils_internal.py
Enable TORCH_TRACE by default in all Tupperware like environments ( #120915 )
2024-03-01 04:47:13 +00:00
_utils.py
[torch.export] Support is_compiling() flag for non-strict mode ( #119602 )
2024-02-29 05:52:51 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
abi-check.cpp
CMakeLists.txt
custom_class_detail.h
custom_class.h
extension.h
functional.py
Fix ouput typos ( #120870 )
2024-02-29 08:29:14 +00:00
hub.py
Add verbose parameter to torch.hub.list ( #120717 )
2024-03-01 07:39:48 +00:00
library.h
library.py
Better error messages for impl_abstract_pystub ( #120959 )
2024-03-04 15:24:36 +00:00
overrides.py
Add assign argument to torch.Tensor.module_load ( #121158 )
2024-03-06 01:32:06 +00:00
py.typed
quasirandom.py
random.py
[2/2] Intel GPU Runtime Upstreaming for Generator ( #118613 )
2024-02-28 05:28:11 +00:00
README.txt
return_types.py
register torch.return_types in torch.fx._pytree ( #120027 )
2024-02-23 21:52:42 +00:00
script.h
serialization.py
Add support to save safetensors checkpoint directly into onnx ( #121001 )
2024-03-11 15:21:59 +00:00
storage.py
Add hpu device support in storage/resize ( #119761 )
2024-02-17 01:04:27 +00:00
torch_version.py
types.py
version.py.tpl