Awni Hannun
|
8c34c9dac4
|
throw for invalid case and remove test (#1575)
|
2024-11-08 12:04:03 -08:00 |
|
Awni Hannun
|
df964132fb
|
fix scatter + test (#1202)
* fix scatter + test
* fix test warnings
* fix metal validation
|
2024-06-11 14:35:12 -07:00 |
|
Alex Barron
|
27d70c7d9d
|
Feature complete Metal FFT (#1102)
* feature complete metal fft
* fix contiguity bug
* jit fft
* simplify rader/bluestein constant computation
* remove kernel/utils.h dep
* remove bf16.h dep
* format
---------
Co-authored-by: Alex Barron <abarron22@apple.com>
|
2024-06-06 12:57:25 -07:00 |
|
Alex Barron
|
2e7c02d5cd
|
Metal FFT for powers of 2 up to 2048 (#915)
* add Metal FFT for powers of 2
* skip GPU test on linux
* fix contiguity bug
* address comments
* Update mlx/backend/metal/fft.cpp
* Update mlx/backend/metal/fft.cpp
* fix bug in synch
---------
Co-authored-by: Alex Barron <abarron22@apple.com>
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
|
2024-04-11 21:40:06 -07:00 |
|
Diogo
|
35431a4ac8
|
Adds device context manager (#679)
|
2024-02-14 14:14:58 -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 |
|