Joona Havukainen
|
9a742090ae
|
Remove tuple unpacking syntax to comply with earlier python versions. Add cuda skip to nanpropagation tests, fix cuda implementation in a separate PR.
|
2025-07-08 20:56:06 -07:00 |
|
Joona Havukainen
|
aca7fac9ef
|
Make the max nanpropagation test more meaningful for integer types
|
2025-07-08 16:42:19 -07:00 |
|
Joona Havukainen
|
e0bb9f3ef8
|
Fix max complex64 nan propagation and add test
|
2025-07-06 15:53:50 -07:00 |
|
Joona Havukainen
|
5b089dc5da
|
Pre-commit formatting
|
2025-07-06 15:50:16 -07:00 |
|
Joona Havukainen
|
af74818528
|
Adding benchmarks and testing for max op nanpropagation
|
2025-07-06 15:50:16 -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 |
|
Awni Hannun
|
0c5eea226b
|
Reduce specializations (#1607)
* start of reduce specializations
* fix all reduce
* fix many dims
* fix
* non-jit tests clear
* cleanup instantiations
* cpu merges
* change dim specializations
* optimize
* fix jit
* fix jit
* use higher precision for integer sum+prod
* fixes
|
2024-11-21 19:53:00 -08:00 |
|
Angelos Katharopoulos
|
969337345f
|
Fix reduce edge case (#1389)
|
2024-09-01 21:37:51 -07:00 |
|
Angelos Katharopoulos
|
8081df79be
|
Fix boolean all reduce bug (#1355)
|
2024-08-24 10:09:32 -07:00 |
|
Angelos Katharopoulos
|
b57a52813b
|
Further reduction tuning (#1349)
* More reduction tuning
* Forgotten pdb
* Small column long row specialization
|
2024-08-23 10:35:25 -07:00 |
|
Awni Hannun
|
98b6ce3460
|
Refactor reductions and fix scatter atomics for large sizes (#1300)
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
|
2024-08-22 16:03:31 -07:00 |
|
Vijay Krish
|
fcc5ac1c64
|
Add GPU support for uint64/int64 reductions (#569)
|
2024-01-31 11:18:04 -08:00 |
|
Awni Hannun
|
b9e415d19c
|
bump pre commit and fix format (#373)
|
2024-01-04 16:28:52 -08:00 |
|
Zach Schillaci
|
5b9be57ac3
|
Add isort pre-commit and run (#68)
|
2023-12-08 11:31:47 -08:00 |
|
Awni Hannun
|
46a39e5b1f
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
Awni Hannun
|
8ca7f9e8e9
|
awni's commit files
|
2023-11-29 10:30:41 -08:00 |
|