add CUDAKernels

mcabbott · mcabbott · commit 6ae8678d306c · 2021-04-04T20:41:33.000-04:00
diff --git a/Project.toml b/Project.toml
@@ -10,6 +10,7 @@ Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 
 [compat]
 CUDA = "1, 2"
+CUDAKernels = "0.1"
 DiffRules = "1"
 FillArrays = "0.10"
 ForwardDiff = "0.10"
@@ -26,6 +27,7 @@ julia = "1.3"
 
 [extras]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+CUDAKernels = "72cfdca4-0801-4ab0-bf6a-d52aa10adc57"
 FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
diff --git a/test/cuda.jl b/test/cuda.jl
@@ -1,6 +1,6 @@
 
 using Tullio, Test
-using CUDA, KernelAbstractions
+using CUDA, CUDAKernels, KernelAbstractions
 CUDA.allowscalar(false)
 using Tracker
 @tullio grad=Base
