Merge pull request #156 from tknopp/main

Switch Compat to be compatible with Flux 0.14

Author: ChrisRackauckas

Project.toml

@@ -5,7 +5,6 @@ version = "0.4.7"
 
 [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"
@@ -17,12 +16,11 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
 CUDA = "3, 4, 5"
-CUDAKernels = "0.3, 0.4"
 ChainRulesCore = "1"
 FFTW = "1"
-Flux = "0.13"
-GeometricFlux = "0.13"
-KernelAbstractions = "0.7, 0.8"
+Flux = "0.13, 0.14"
+GeometricFlux = "0.13, 0.14"
+KernelAbstractions = "0.7, 0.8, 0.9"
 Statistics = "1"
 Tullio = "0.3"
 Zygote = "0.6"

src/NeuralOperators.jl

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

src/Transform/fourier_transform.jl

@@ -19,5 +19,5 @@ truncate_modes(ft::FourierTransform, 𝐱_fft::AbstractArray) = low_pass(ft, 
 
 function inverse(ft::FourierTransform, 𝐱_fft::AbstractArray{T, N},
                  M::NTuple{N, Int64}) where {T, N}
-    return real(irfft(𝐱_fft, M[1], 1:ndims(ft))) # [size(x_fft)..., out_chs, batch]
+    return real(irfft(complex(𝐱_fft), M[1], 1:ndims(ft))) # [size(x_fft)..., out_chs, batch]
 end