[WIP] See if we can make symbolic values default - #2795

Draft

shino16 wants to merge 47 commits into

mainfrom

make-symbolic-default

Commits on Dec 10, 2025

Register methods to NumberProxy
shino16
committed
9d4110e
Register Python's scalar ops for pythonex
shino16
committed
23a677b
Python's floor/ceil/round/trunc returns int
shino16
committed
68a5bcc
Support all-number inputs to where, register to pythonex
shino16
committed
88d971f
Add tests, recompile when number type changes
shino16
committed
0d4d1db
Add test_where_on_numbers
shino16
committed
40f4e7e
Reduce test time
shino16
committed
13a2d01
Do not cache ComplexProxy, test on where
shino16
committed
6c925fa
Reflect INT_FOR_NUMBER behavior in nvfuserex
shino16
committed
135a835
Add py_min, py_max
shino16
committed
592b3de
Add torch.sym_min/max
shino16
committed
c55ac4a
Add test
shino16
committed
522b3f5
Add builtin min/max lookaside
shino16
committed
3b73a42
Add test
shino16
committed
e2578e3
Remove py_max/min and merge into prims.maximum

shino16
and
riccardofelluga
committed
df72dad
Fix up
shino16
committed
2806c40
Apply DCE to subsymbols

beverlylytle
authored and
shino16
committed
37bfd92
try in symbol.__call__ instead

beverlylytle
authored and
shino16
committed
5261098
come on, ruff, it's a test

beverlylytle
authored and
shino16
committed
681492f
remove print

beverlylytle
authored and
shino16
committed
6af4d20
respond to comments

beverlylytle
authored and
shino16
committed
10d0564
where's my coffee

beverlylytle
authored and
shino16
committed
2b7d2b3
MAKE SYMBOLIC VALUES DEFAULT
shino16
committed
d9e4f9c
Shift bsym indices
shino16
committed
8293bd5
Add torch.set_autocast_enabled
shino16
committed
2e0015f
Add StringProxy.__bool__
shino16
committed
211a2a4
Defer cotangents creation after forward pass
shino16
committed
97a0032
.numel needs a preceding prims.shape
shino16
committed
afb6a5a
Prologue can't be skipped
shino16
committed
77d15b4
Support set_grad_enabled(symbolic)
shino16
committed
a262150
Support isinstance(symbolic, type)
shino16
committed
b9df532
a.numel() instead of a.numel
shino16
committed
06baafb
Remove _take_check numel check
shino16
committed
ec8ce20
FIXME Allow a cycle in swap_map in OpExProcessor
shino16
committed
b965ef9
Adjust test
shino16
committed
ae5806e
Adjust test
shino16
committed
4b8312f
Adjust test
shino16
committed
a78049d
Adjust test
shino16
committed
b1013ba
Remove obsolete xfail mark
shino16
committed
dab3ad2
Do not treat str as symbolic values
shino16
committed
18d34b3
Adjust test
shino16
committed
634de48
Skip test
shino16
committed
c668c1c
Avoid len(a.shape) in executor checker
shino16
committed
829074c

Commits on Dec 11, 2025

Do not treat bool inputs symbolically
shino16
committed
d747849
Adjust test
shino16
committed
fa642ac
Update test
shino16
committed
172320d

Commits on Dec 15, 2025

Merge branch 'main' of ssh://github.com/Lightning-AI/lightning-thunder into make-symbolic-default
shino16
committed
b3a78f1