From 5d7efafe92be7b6083819946fc9103ab3f400d0e Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Wed, 22 Oct 2025 14:46:44 -0700
Subject: [PATCH] fix compile

---
 mlx/backend/cuda/device/unary_ops.cuh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mlx/backend/cuda/device/unary_ops.cuh b/mlx/backend/cuda/device/unary_ops.cuh
index e15d1cf52..61c587e27 100644
--- a/mlx/backend/cuda/device/unary_ops.cuh
+++ b/mlx/backend/cuda/device/unary_ops.cuh
@@ -2,6 +2,8 @@
 
 #pragma once
 
+#include <cuda_fp8.h>
+
 #include "mlx/backend/cuda/device/fp16_math.cuh"
 #include "mlx/backend/cuda/device/utils.cuh"