chenyu and GitHub
ba4b8c2c23
Tensor.copysign ( #9329 )
2025-03-02 21:33:49 -05:00
Friedrich Carl Eichenroth and GitHub
06ef9cc9f4
aten leaky_relu, div.out_mode, clamp_max, clamp_min, copysign ( #9323 )
...
* fix some torch tests
* fixup
* small change
* fixup
* fix test
* use default function
* add todo
2025-03-02 19:12:16 -05:00
Priyank Patel and GitHub
f4148ac46a
torch fix casting and add ops for sd vae(s) ( #9297 )
...
* torch fix copy casting and add upsample op
* update cast and add test
* fix lint
* add pad for sdxl vae to work
2025-03-01 08:49:10 -05:00
nimlgen and GitHub
052722a7bc
torch hook: address comments ( #9295 )
...
* torch hook: address comments
* failed test
2025-02-28 11:51:52 +03:00
Priyank Patel and GitHub
8ae215dd3d
torch backend fix manual seed warning ( #9292 )
2025-02-28 13:45:32 +08:00
geohot
ac40316692
hotfix: group cpu functions in torch backend
2025-02-28 10:39:00 +08:00
George Hotz and GitHub
b32595dbbc
torch examples ( #9290 )
...
* torch, fix examples/mnist
* fix vae torch example
* where out
2025-02-28 10:16:06 +08:00
chenyu and GitHub
184030168d
fix aten.reflection_pad2d ( #9289 )
...
tested the torch doc example
2025-02-27 15:53:46 -05:00
chenyu and GitHub
0de6585df0
fix aten.normal_ arg ( #9288 )
...
should be mean and std.
2025-02-27 15:36:25 -05:00
chenyu and GitHub
8ee2b460ee
Tensor.var_mean ( #9287 )
2025-02-27 15:15:31 -05:00
George Hotz and GitHub
387ea41e99
increase speed of torch mnist: use gradient api ( #9282 )
2025-02-27 11:57:41 +08:00
a0764f0dc0
(bounty) Make mnist training run with torch backend ( #9233 )
...
* yml changes
* torch backend remove meta decomps and add test
* torch backend bump timeout for tests
---------
Co-authored-by: George Hotz <[email protected]
2025-02-27 11:32:25 +08:00
George Hotz and GitHub
9088125a6a
a lil more torch ( #9280 )
2025-02-27 11:12:20 +08:00
George Hotz and GitHub
b6a14911c8
start torch.compile support ( #9279 )
2025-02-27 10:29:51 +08:00
Francis Lata and GitHub
86b737a120
leakyrelu to leaky_relu ( #9270 )
2025-02-26 13:22:08 -05:00
chenyu and GitHub
aaf0a8069f
xor -> bitwise_xor ( #9264 )
2025-02-26 10:21:14 -05:00
George Hotz and GitHub
2158dc4849
full fix for as_strided in torch backend ( #9257 )
...
* fixes from chargpt for torch backend
* shrink support
* add stride support
* comment cleanup
* a few more
* work
* import the stream hack
* llvm multi auto
2025-02-26 22:34:05 +08:00
George Hotz and GitHub
7780393460
rig up torch's testing framework [pr] ( #9254 )
...
* rig up torch's testing framework [pr]
* support more movement ops
* dec on expand
* fix tests
* work
* fix tests
* a few more
* decomps + opt hook
* installed pytest
2025-02-26 18:46:22 +08:00
George Hotz and GitHub
b603af373e
run some tests from torch [pr] ( #9252 )
...
* run some tests from torch [pr]
* yml
* wrap_out
* clean up for the new people
* a lil more
2025-02-26 15:42:22 +08:00
George Hotz and GitHub
fc32ff80d6
torch and numpy dtype interop [pr] ( #9224 )
...
* torch and numpy dtype interop [pr]
* less lines
* order
2025-02-24 18:26:49 +08:00
geohot
fd731e740a
hotfix: add note on backend2.py
2025-02-24 11:23:03 +08:00
albanD and GitHub
f2dd9c1562
simplify c++ code ( #9221 )
2025-02-24 11:04:41 +08:00
George Hotz and GitHub
97bc723538
torch backend works for ResNet-18 ( #9200 )
...
* torch backend progress, a few more functions
* resnet works
* pillow
* tv
2025-02-22 22:16:23 +08:00
George Hotz and GitHub
4e6665bda5
different way to write torch backend ( #9197 )
...
* different way to write torch backend
* both backends
* more work
* simpler code
* more work
* test both
* imply unwrap/wrap
* FORWARD_ONLY=1 TINY_BACKEND=1 python3 test/test_ops.py TestOps.test_add works
* ready to start making test_ops work in torch backend
* backward pass, TINY_BACKEND=1 python3 test/test_ops.py TestOps.test_add works
* FORWARD_ONLY=1 TINY_BACKEND=1 python3 test/test_ops.py TestOps.test_simple_conv2d works
* matmul backward is broken with as_strided
2025-02-22 14:42:26 +08:00
George Hotz and GitHub
e87be0131e
torch backend start ( #9191 )
...
* start torch backend
* progress
* ugh, you need cpp crap
* 1+1 works
* 1+1 works
* becoming a real backend
* ready to merge?
2025-02-21 16:57:28 +08:00
geohot