diff --git a/Project.toml b/Project.toml
index 080186b..109c069 100644
--- a/Project.toml
+++ b/Project.toml
@@ -22,7 +22,7 @@ ChainRulesCore = "1"
 FFTW = "1"
 Flux = "0.13"
 GeometricFlux = "0.13"
-KernelAbstractions = "0.7, 0.8"
+KernelAbstractions = "0.7, 0.8, 0.9"
 Tullio = "0.3"
 Zygote = "0.6"
 julia = "1.6"