From 4d485fca24b3a055ca47a3f42e7d65a5b5bd397c Mon Sep 17 00:00:00 2001
From: Alex Barron <barronalex@users.noreply.github.com>
Date: Mon, 3 Jun 2024 09:50:10 -0700
Subject: [PATCH] Add defines include (#1176)

Co-authored-by: Alex Barron <abarron22@apple.com>
---
 mlx/backend/metal/kernels/scaled_dot_product_attention.metal | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mlx/backend/metal/kernels/scaled_dot_product_attention.metal b/mlx/backend/metal/kernels/scaled_dot_product_attention.metal
index a7cccc710..8d3506c34 100644
--- a/mlx/backend/metal/kernels/scaled_dot_product_attention.metal
+++ b/mlx/backend/metal/kernels/scaled_dot_product_attention.metal
@@ -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"