Index _ | A | B | C | D | E | F | G | I | J | L | M | N | P | R | S | T | U | V | W | Z _ __call__() (thunder.core.langctxs.langctx method) (thunder.core.symbol.Symbol method) (thunder.torch.torchsymbol method) __init__() (thunder.common.CACHE_OPTIONS method) (thunder.core.baseutils.BoundSymbolInterface method) (thunder.core.baseutils.NumberProxyInterface method) (thunder.core.baseutils.ProxyInterface method) (thunder.core.baseutils.SymbolInterface method) (thunder.core.baseutils.TensorProxyInterface method) (thunder.core.codeutils.SigInfo method) (thunder.core.devices.Device method) (thunder.core.devices.DeviceType method) (thunder.core.dtypes.dtype method) (thunder.core.langctxs.langctx method) (thunder.core.prims.PrimIDs method) (thunder.core.proxies.CollectionProxy method) (thunder.core.proxies.ComplexProxy method) (thunder.core.proxies.FloatProxy method) (thunder.core.proxies.FutureTensorProxy method) (thunder.core.proxies.IntegerProxy method) (thunder.core.proxies.TensorProxy method) (thunder.core.proxies.Variable method) (thunder.core.symbol.BoundSymbol method) (thunder.core.symbol.BoundSymbolRHS method) (thunder.core.symbol.Symbol method) (thunder.core.trace.TraceCtx method) (thunder.core.transforms.Node method) (thunder.core.transforms.VISIT_TYPE method) (thunder.distributed.FSDPBucketingStrategy method) (thunder.distributed.FSDPType method) (thunder.executors.torch_compile.TorchCompileExecutor method) (thunder.executors.utils.Region method) (thunder.torch.torchsymbol method) A a (thunder.clang.matrix_transpose parameter), [1] abs() (in module thunder.clang) acos() (in module thunder.clang) acosh() (in module thunder.clang) add() (in module thunder.clang) apex_available() (in module thunder.executors.apex_entropyex) arange() (in module thunder.clang) are_all_args_constant (thunder.core.symbol.BoundSymbol property) asin() (in module thunder.clang) asinh() (in module thunder.clang) atan() (in module thunder.clang) atan2() (in module thunder.clang) atanh() (in module thunder.clang) B bfloat16 (in module thunder.core.dtypes) bitwise_and() (in module thunder.clang) bitwise_not() (in module thunder.clang) bitwise_or() (in module thunder.clang) bitwise_xor() (in module thunder.clang) BLOCK (thunder.distributed.FSDPBucketingStrategy attribute) bool8 (in module thunder.core.dtypes) BoundSymbol (class in thunder.core.symbol) BoundSymbolInterface (class in thunder.core.baseutils) BoundSymbolRHS (class in thunder.core.symbol) broadcast_from (thunder.distributed.ddp parameter), [1] (thunder.distributed.fsdp parameter), [1] bsym (thunder.executors.torch_compile.to_torch_translator parameter), [1] bsym_list_to_dag() (in module thunder.core.transforms) bucket_size_in_mb (thunder.distributed.ddp parameter), [1] bucketing_strategy (thunder.distributed.fsdp parameter), [1] C cache (thunder.functional.jit parameter), [1] (thunder.jit parameter), [1] cache_hits() (in module thunder) cache_misses() (in module thunder) cache_option() (in module thunder) CACHE_OPTIONS (class in thunder.common) cat() (in module thunder.clang) ceil() (in module thunder.clang) check() (in module thunder.core.baseutils) check_type() (in module thunder.core.baseutils) clear_collections (thunder.executors.passes.del_last_used parameter), [1] CollectionProxy (class in thunder.core.proxies) compile_data() (in module thunder) compile_stats() (in module thunder) complex128 (in module thunder.core.dtypes) complex32 (in module thunder.core.dtypes) complex64 (in module thunder.core.dtypes) ComplexProxy (class in thunder.core.proxies) compute_broadcast_shape() (in module thunder.clang) convolution() (in module thunder.clang) copysign() (in module thunder.clang) cos() (in module thunder.clang) cosh() (in module thunder.clang) cudnn_ex (in module thunder.executors.cudnnex) D ddp() (in module thunder.distributed) del_last_used() (in module thunder.executors.passes) Device (class in thunder.core.devices) device (thunder.distributed.fsdp parameter), [1] device_from_string() (in module thunder.core.devices) device_put() (in module thunder.clang) DeviceType (class in thunder.core.devices) devicetype_string() (in module thunder.core.devices) diagonal() (in module thunder.clang) dtype (class in thunder.core.dtypes) E early_transforms (thunder.jit parameter), [1] eq() (in module thunder.clang) erf() (in module thunder.clang) erfc() (in module thunder.clang) erfcinv() (in module thunder.clang) erfinv() (in module thunder.clang) examine() (in module thunder.examine) executors (thunder.functional.jit parameter), [1] (thunder.jit parameter), [1] exp() (in module thunder.clang) exp2() (in module thunder.clang) expand() (in module thunder.clang) expm1() (in module thunder.clang) F flatten() (in module thunder.clang) float16 (in module thunder.core.dtypes) float32 (in module thunder.core.dtypes) float64 (in module thunder.core.dtypes) FloatProxy (class in thunder.core.proxies) floor() (in module thunder.clang) floor_divide() (in module thunder.clang) fmod() (in module thunder.clang) fn (thunder.functional.jit parameter), [1] (thunder.jit parameter), [1] from_bsym_swap_proxies() (thunder.core.symbol.BoundSymbol method) from_trace() (in module thunder.core.trace) fsdp() (in module thunder.distributed) FSDPBucketingStrategy (class in thunder.distributed) FSDPType (class in thunder.distributed) full() (in module thunder.clang) full_like() (in module thunder.clang) FutureTensorProxy (class in thunder.core.proxies) G ge() (in module thunder.clang) get_buffer() (thunder.ThunderModule method) get_langctx() (in module thunder.core.langctxs) get_nvfuser_executor() (in module thunder.executors) get_parameter() (thunder.ThunderModule method) get_siginfo() (in module thunder.core.codeutils) get_skip_data_parallel_grad_sync() (in module thunder.distributed) get_submodule() (thunder.ThunderModule method) get_torch_executor() (in module thunder.executors) get_tracectx() (in module thunder.core.trace) gt() (in module thunder.clang) I index_add() (in module thunder.clang) insert_inplace() (in module thunder.core.transforms) int16 (in module thunder.core.dtypes) int32 (in module thunder.core.dtypes) int64 (in module thunder.core.dtypes) int8 (in module thunder.core.dtypes) IntegerProxy (class in thunder.core.proxies) interpretation (thunder.functional.jit parameter), [1] (thunder.jit parameter), [1] is_proxyable() (in module thunder.core.proxies) isfinite() (in module thunder.clang) J jit() (in module thunder) (in module thunder.functional) L langctx (class in thunder.core.langctxs) (thunder.functional.jit parameter), [1] (thunder.jit parameter), [1] last_backward_traces() (in module thunder) last_compile_options() (in module thunder) last_interpreted_instructions() (in module thunder) last_interpreter_log() (in module thunder) last_prologue_traces() (in module thunder) last_traces() (in module thunder) LAYER (thunder.distributed.FSDPBucketingStrategy attribute) le() (in module thunder.clang) lgamma() (in module thunder.clang) list_transforms() (in module thunder) log() (in module thunder.clang) log10() (in module thunder.clang) log1p() (in module thunder.clang) log2() (in module thunder.clang) logical_and() (in module thunder.clang) lt() (in module thunder.clang) M make_compiled() (in module thunder.executors.torch_compile) matrix_transpose() (in module thunder.clang) maybe_broadcast() (in module thunder.clang) maybe_convert_to_dtype() (in module thunder.clang) mod() (in module thunder.clang) model (thunder.distributed.ddp parameter), [1] module thunder thunder.clang thunder.common thunder.core thunder.core.codeutils thunder.core.devices thunder.core.dtypes thunder.core.langctxs thunder.core.prims thunder.core.proxies thunder.core.pytree thunder.core.symbol thunder.core.trace thunder.core.transforms thunder.distributed thunder.examine thunder.executors thunder.executors.apex_entropyex thunder.executors.cudnnex thunder.executors.nvfuserex thunder.executors.passes thunder.executors.pythonex thunder.executors.torch_compile thunder.executors.torchex thunder.executors.triton_crossentropy thunder.executors.utils thunder.torch thundre.core.rematerialization movedim() (in module thunder.clang) mul() (in module thunder.clang) N ndtri() (in module thunder.clang) ne() (in module thunder.clang) neg() (in module thunder.clang) nextafter() (in module thunder.clang) no_sync() (thunder.ThunderModule method) Node (class in thunder.core.transforms) NONE (thunder.distributed.FSDPBucketingStrategy attribute) NumberProxyInterface (class in thunder.core.baseutils) nvfuser_available() (in module thunder.executors) P pow() (in module thunder.clang) prettyprint() (in module thunder.core.codeutils) PrimIDs (class in thunder.core.prims) proxy() (in module thunder.core.proxies) ProxyInterface (class in thunder.core.baseutils) pytype() (in module thunder.core.proxies) pyval() (in module thunder.core.proxies) R reciprocal() (in module thunder.clang) Region (class in thunder.executors.utils) remainder() (in module thunder.clang) rematerialize() (in module thunder.core.rematerialization) replace_inplace() (in module thunder.core.transforms) replace_name() (thunder.core.proxies.ComplexProxy method) (thunder.core.proxies.FloatProxy method) (thunder.core.proxies.FutureTensorProxy method) (thunder.core.proxies.IntegerProxy method) (thunder.core.proxies.TensorProxy method) reset_langctx() (in module thunder.core.langctxs) reset_skip_data_parallel_grad_sync() (in module thunder.distributed) reset_tracectx() (in module thunder.core.trace) reshape() (in module thunder.clang) round() (in module thunder.clang) rsqrt() (in module thunder.clang) S scatter_add() (in module thunder.clang) set_langctx() (in module thunder.core.langctxs) set_skip_data_parallel_grad_sync() (in module thunder.distributed) set_tracectx() (in module thunder.core.trace) sharding_strategy (thunder.distributed.fsdp parameter), [1] sharp_edges (thunder.functional.jit parameter), [1] (thunder.jit parameter), [1] SigInfo (class in thunder.core.codeutils) sigmoid() (in module thunder.clang) sign() (in module thunder.clang) signbit() (in module thunder.clang) silu() (in module thunder.clang) sin() (in module thunder.clang) sinh() (in module thunder.clang) size() (in module thunder.torch) skip_data_parallel_grad_sync() (in module thunder.distributed) skip_inputs (thunder.core.symbol.BoundSymbol.from_bsym_swap_proxies parameter), [1] skip_output (thunder.core.symbol.BoundSymbol.from_bsym_swap_proxies parameter), [1] skip_subsymbols (thunder.core.symbol.BoundSymbol.from_bsym_swap_proxies parameter), [1] slice_in_dim() (in module thunder.clang) sqrt() (in module thunder.clang) squeeze() (in module thunder.clang) stack() (in module thunder.clang) stride_order() (in module thunder.clang) sub() (in module thunder.clang) Symbol (class in thunder.core.symbol) SymbolInterface (class in thunder.core.baseutils) T take() (in module thunder.clang) take_along_axis() (in module thunder.clang) tan() (in module thunder.clang) tanh() (in module thunder.clang) target (thunder.ThunderModule.get_buffer parameter) (thunder.ThunderModule.get_parameter parameter) (thunder.ThunderModule.get_submodule parameter) TensorProxy (class in thunder.core.proxies) TensorProxyInterface (class in thunder.core.baseutils) thunder module thunder.clang module thunder.common module thunder.core module thunder.core.codeutils module thunder.core.devices module thunder.core.dtypes module thunder.core.langctxs module thunder.core.prims module thunder.core.proxies module thunder.core.pytree module thunder.core.symbol module thunder.core.trace module thunder.core.transforms module thunder.distributed module thunder.examine module thunder.executors module thunder.executors.apex_entropyex module thunder.executors.cudnnex module thunder.executors.nvfuserex module thunder.executors.passes module thunder.executors.pythonex module thunder.executors.torch_compile module thunder.executors.torchex module thunder.executors.triton_crossentropy module thunder.executors.utils module thunder.torch module ThunderModule (class in thunder) thundre.core.rematerialization module to_device() (in module thunder.core.devices) to_dtype() (in module thunder.core.dtypes) to_float() (in module thunder.torch) to_torch_translator() (in module thunder.executors.torch_compile) token (thunder.distributed.reset_skip_data_parallel_grad_sync parameter), [1] toposort_bsym_dag() (in module thunder.core.transforms) TorchCompileExecutor (class in thunder.executors.torch_compile) torchsymbol (class in thunder.torch) trace (thunder.core.rematerialization.rematerialize parameter), [1] (thunder.executors.passes.del_last_used parameter), [1] (thunder.executors.passes.update_fusion_call_ctx parameter), [1] TraceCtx (class in thunder.core.trace) transform_for_execution() (in module thunder.executors.passes) transforms (thunder.jit parameter), [1] transpose() (in module thunder.clang) true_divide() (in module thunder.clang) trunc() (in module thunder.clang) U uint8 (in module thunder.core.dtypes) uniform() (in module thunder.clang) uniform_like() (in module thunder.clang) unsqueeze() (in module thunder.clang) unvariableify() (in module thunder.core.proxies) update_fusion_call_ctx() (in module thunder.executors.passes) V value (thunder.distributed.set_skip_data_parallel_grad_sync parameter), [1] Variable (class in thunder.core.proxies) variableify() (in module thunder.core.proxies) VISIT_TYPE (class in thunder.core.transforms) visitor_transform() (in module thunder.core.transforms) W where() (in module thunder.clang) Z ZERO2 (thunder.distributed.FSDPType attribute) ZERO3 (thunder.distributed.FSDPType attribute)