L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add defines include (#1176)

Browse Source 
Co-authored-by: Alex Barron <[email protected]>
This commit is contained in:
Alex Barron
2024-06-03 09:50:10 -07:00
committed by GitHub
co-authored by Alex Barron
parent 1865299a30
commit 4d485fca24
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mlx/backend/metal/kernels/scaled_dot_product_attention.metal
+1
View File
@@ -1,6 +1,7 @@
#include <metal_simdgroup>
#include <metal_stdlib>
#include "mlx/backend/metal/kernels/steel/defines.h"
#include "mlx/backend/metal/kernels/steel/gemm/transforms.h"
#include "mlx/backend/metal/kernels/steel/utils.h"