George Hotz
a0cb16ac61
node cleanup + local metal test speed [pr] ( #6880 )
...
* node cleanup [pr]
* fix tests, including the double one on metal
* no time tqdm tests
2024-10-04 18:14:23 +08:00
George Hotz
b1c0d8c99d
remove cpu and torch backends ( #3399 )
...
* remove cpu and torch backends
* don't copy to cpu
* use clang instead of cpu
* multitensor gathers on the first device
* clang is cpu + use default
* fixup
* bugfix
2024-02-15 16:55:39 +01:00
Oleg Rybalko
c3133adb8c
Disk shm refactor ( #2912 )
...
* better support for platform dependent flags
* osx test support
* removed unused import and made line length <150
* changed osx ci shm
* lstrip in case SharedMemory._name is passed
2023-12-22 09:23:37 -08:00
qazal
ab2d4d8d29
Fix cl import in the copy_speed test and cifar example ( #2586 )
...
* fix CL import
* update test to only run on GPU
* update hlb_cifar too
2023-12-03 09:22:07 -08:00
chenyu
3eb3c74675
metal ci tests everything ( #2499 )
...
* metal ci tests everything
* pretty good
* METAL
2023-11-29 12:04:37 -05:00
Christopher Mauri Milan
7f01dd04f0
Apply ruff linting rules to tests ( #2473 )
...
* everything except F821
* enable F821 with noqa
* dumb fix
* fix remaining imports and (former) lambdas
* replace _ with noqa to avoid gc
2023-11-27 21:24:06 -08:00
George Hotz
9e07824542
move device to device.py ( #2466 )
...
* move device to device.py
* pylint test --disable R,C,W,E --enable E0611
* fix tests
2023-11-27 11:34:37 -08:00
George Hotz
4f8f0ac139
minor cleanups, remove dead files ( #2398 )
...
* minor cleanups, remove dead files
* s.name
* use disk
* pytest passes on mac
2023-11-23 09:01:50 -08:00
George Hotz
cbb8486779
ResNet training changes (update benchmark) ( #2390 )
...
* default arg for chunk
* bring back to_
* good changes
* new set
* unused hash
* fix optim
* new torch loader
* fix test lr scheduler
2023-11-22 17:41:12 -08:00
