pytorch/torch
Colin L. Rice abc5d59dcb config: create Config objects with JK support (#138766)
This teaches install_config_module (and the underlying code) to
understands Config objects. Additionally we've added a JK option to this
which resolves the JK.

This config gets stored within the _ConfigEntry class and is evaluated
when __getattr__ is called. If justknobs is set, it'll call
justknobs_check to see the result.

Due to preceeding work, basically everything works correctly here and we
had to update a couple of tests, and modify the getattr behaviour.

Note that we are updating the justknob_check function to support a
default option, to make default work.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/138766
Approved by: https://github.com/ezyang
2024-11-01 19:20:37 +00:00
..
_awaits
_C [Profiler] Create Auto-Trace Frontend for Trace ID (#139310) 2024-10-31 19:02:57 +00:00
_C_flatbuffer
_custom_op
_decomp addmm: error on output dtype mismatch. (#138520) 2024-10-30 21:46:39 +00:00
_dispatch
_dynamo [dynamo] ignore False/None callback in fail_on_recompile/force_backend stances (#139215) 2024-11-01 06:15:28 +00:00
_export reland D65167805 "[export] Update min_val and max_val to Optional[int] in serialization." (#139394) 2024-10-31 22:28:32 +00:00
_functorch Update tensorify pass to specialize symfloats we didn't tensorify away (#138868) 2024-11-01 03:18:02 +00:00
_higher_order_ops [cond] make cond not throw warnings on constant pred in eager mode (#138837) 2024-10-31 01:13:19 +00:00
_inductor [fx graph cache] Support freezing with FX graph caching (#136505) 2024-11-01 18:29:29 +00:00
_lazy
_library [real_tensor_prop] Infer Fake kernels during real tensor prop (#139213) 2024-10-30 21:29:33 +00:00
_logging [draft-export] Include guards for constraint violation errors (#138748) 2024-10-30 00:24:17 +00:00
_numpy
_prims Revert "Fix unbind_copy and add its decomposition (#134319)" 2024-10-29 04:54:37 +00:00
_prims_common Revert "Fix unbind_copy and add its decomposition (#134319)" 2024-10-29 04:54:37 +00:00
_refs Revert "Fix unbind_copy and add its decomposition (#134319)" 2024-10-29 04:54:37 +00:00
_strobelight Increase default COMPILE_STROBELIGHT_MAX_STACK_LENGTH to 500 (#138006) 2024-10-17 07:31:32 +00:00
_subclasses [export] Initial draft export (#139383) 2024-11-01 06:25:44 +00:00
_vendor
accelerator Introduce a device-agnostic runtime API design (#132204) 2024-10-27 10:37:09 +00:00
amp [MPS] Update error message for supported autocast type (#139192) 2024-10-30 16:48:29 +00:00
ao [AOTI] Use len(serialized_weights) when calculating consts_size (#139054) 2024-10-31 09:54:16 +00:00
autograd [Profiler] Create Auto-Trace Frontend for Trace ID (#139310) 2024-10-31 19:02:57 +00:00
backends Revert "[sparse] add search for optimal alg_id to torch.compile (#137427)" 2024-10-24 17:27:06 +00:00
compiler [dynamo] add compiler.set_stance raw function call test and doc example (#138276) 2024-10-18 02:54:22 +00:00
contrib Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
cpu Extend vectorization with SVE(ARM) with Torch Compile (Inductor) (#134672) 2024-10-10 13:20:40 +00:00
csrc [pgnccl][simple] clean up unused members of PGNCCL (#139436) 2024-11-01 16:25:04 +00:00
cuda Adds snapshot API for MemPools to get pool memory segments (#133601) 2024-10-29 01:01:47 +00:00
distributed [Pipelining] Update schedules to use I, B actions. (#138886) 2024-11-01 03:54:06 +00:00
distributions Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
export [export] Initial draft export (#139383) 2024-11-01 06:25:44 +00:00
fft
func
futures
fx [Partitioner] Enumerate partitions by iterating partition ids (#136598) 2024-11-01 07:42:36 +00:00
jit Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
legacy
lib
linalg
masked [BE]: Update Typeguard to TypeIs for better type inference (#133814) 2024-10-26 15:07:13 +00:00
monitor
mps
mtia
multiprocessing Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
nested Fix and test several NJT reductions (#139317) 2024-10-31 20:55:38 +00:00
nn Add Weighted Loss Functions to PyTorch : WMSE, WMAE, and Weighted Huber Loss (#132049) 2024-10-31 21:59:43 +00:00
onnx [FX][export][dynamo] use tuple instead of list in normalized args_spec (#138212) 2024-10-25 06:43:55 +00:00
optim Ensure SWA boundary conditions w.r.t. definition (#133773) 2024-10-31 18:24:08 +00:00
package Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
profiler [fx graph cache] Support freezing with FX graph caching (#136505) 2024-11-01 18:29:29 +00:00
quantization Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
signal
sparse Revert "[sparse] add search for optimal alg_id to torch.compile (#137427)" 2024-10-24 17:27:06 +00:00
special
testing config: create Config objects with JK support (#138766) 2024-11-01 19:20:37 +00:00
utils config: create Config objects with JK support (#138766) 2024-11-01 19:20:37 +00:00
xpu Add torch.xpu.get_arch_list and torch.xpu.get_gencode_flags for XPU (#137773) 2024-10-18 02:28:08 +00:00
__config__.py
__future__.py
__init__.py Add conjugate method on SymFloat (#139249) 2024-10-31 04:55:36 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py
_deploy.py
_environment.py
_guards.py [hierarchical-compilation][invoke_subgraph] Use tracing context to cache artifacts of dispatch keys (#137965) 2024-10-22 15:33:42 +00:00
_jit_internal.py
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py Revert "[sparse] add search for optimal alg_id to torch.compile (#137427)" 2024-10-24 17:27:06 +00:00
_namedtensor_internals.py
_ops.py remove redundant a (#139046) 2024-10-28 17:47:24 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor.py Remove numpy dependency for maia serialization (#137600) 2024-10-28 20:57:35 +00:00
_tensor_docs.py
_tensor_str.py
_thread_safe_fork.py
_torch_docs.py make equation behind torch.isclose element-wise (#138459) 2024-11-01 18:18:33 +00:00
_utils.py Fix .to(cpu) for Storage (#138011) 2024-10-23 01:31:48 +00:00
_utils_internal.py config: create Config objects with JK support (#138766) 2024-11-01 19:20:37 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py Fix weights_only for BUILD instructions for user allowlisted objects with __slots__ (#138936) 2024-11-01 00:59:29 +00:00
abi-check.cpp
CMakeLists.txt Add torch.xpu.get_arch_list and torch.xpu.get_gencode_flags for XPU (#137773) 2024-10-18 02:28:08 +00:00
custom_class.h Remove some pre-cpp17 stuff (#138410) 2024-10-23 00:38:03 +00:00
custom_class_detail.h Remove some pre-cpp17 stuff (#138410) 2024-10-23 00:38:03 +00:00
extension.h
functional.py Clarify opt-einsum usage, fix #127109 (#137596) 2024-10-09 20:31:24 +00:00
hub.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
library.h [1/N] Enable cppcoreguidelines-special-member-functions (#137405) 2024-10-23 00:16:53 +00:00
library.py Fix custom op bug of clearing dir (#137655) 2024-10-11 04:32:40 +00:00
overrides.py Add Weighted Loss Functions to PyTorch : WMSE, WMAE, and Weighted Huber Loss (#132049) 2024-10-31 21:59:43 +00:00
py.typed
quasirandom.py
random.py [Torch] Support meta device in random.fork_rng (#137715) 2024-10-16 18:00:39 +00:00
README.txt
return_types.py
script.h
serialization.py reimport pr137735 due to merging check issues (#138959) 2024-10-27 16:31:34 +00:00
storage.py Fix .to(cpu) for Storage (#138011) 2024-10-23 01:31:48 +00:00
torch_version.py
types.py
version.py.tpl

Note [TH abstraction violation]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

TH/THC provide some hpp headers, which are proper C++ headers rather than
C headers.  These headers serve double duty as *internal implementation
detail* headers, whose contents should largely not be used by external
clients.

Ideally, we would not install these headers at all; instead, you should
use public functions (in headers like `THTensor.h`, NOT `THTensor.hpp`)
to manipulate these structs.  However, there are a few places
in torch/csrc where we violate this abstraction.  They are marked with
a pointer to this note.  Each of those sites will have to be refactored
when we refactor the guts of THTensor and related structures.