Angelos Katharopoulos
|
fd1d0821d2
|
Make sure softmax doesn't change the actual max
|
2025-06-22 23:34:32 -07:00 |
|
Awni Hannun
|
cad5c0241c
|
[CUDA] synch properly waits for all tasks to finish and clear (#2303)
* cuda synch properly waits for all tasks to finish and clear
* fix copy
|
2025-06-17 12:03:25 -07:00 |
|
Awni Hannun
|
b8022c578a
|
divmod, partition, sort fixes (#2302)
|
2025-06-16 18:49:32 -07:00 |
|
Awni Hannun
|
bc53f8293f
|
Cuda bug fixes 2 (#2298)
* more bug fixes
* more bug fixes
* format
|
2025-06-16 13:14:46 -07:00 |
|
Awni Hannun
|
c552ff2451
|
[CUDA] Fix back-end bugs and enable corresponding tests (#2296)
* Fix some cuda back-end bugs and enable corresponding tests
* more fixes
* enable more tests
* format
|
2025-06-16 08:45:40 -07:00 |
|
Awni Hannun
|
4fda5fbdf9
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|