change to Flux version 0.14

Author: tknopp

Project.toml

@@ -5,7 +5,6 @@ version = "0.4.6"
 
 [deps]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
-CUDAKernels = "72cfdca4-0801-4ab0-bf6a-d52aa10adc57"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 FFTW = "7a1cc6ca-52ef-59f5-83cd-3a7055c09341"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
@@ -16,13 +15,12 @@ Tullio = "bc48ee85-29a4-5162-ae0b-a64e1601d4bc"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
-CUDA = "3, 4"
-CUDAKernels = "0.3, 0.4"
+CUDA = "3, 4, 5"
 ChainRulesCore = "1"
 FFTW = "1"
-Flux = "0.13"
-GeometricFlux = "0.13"
-KernelAbstractions = "0.7, 0.8"
+Flux = "0.14"
+GeometricFlux = "0.14"
+KernelAbstractions = "0.7, 0.8, 0.9"
 Tullio = "0.3"
 Zygote = "0.6"
 julia = "1.6"

src/NeuralOperators.jl

@@ -4,7 +4,6 @@ using Flux
 using FFTW
 using Tullio
 using CUDA
-using CUDAKernels
 using KernelAbstractions
 using Zygote
 using ChainRulesCore