Add EnzymeCore extension

Author: wsmoses

lib/GPUArraysCore/Project.toml

@@ -6,6 +6,13 @@ version = "0.1.6"
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 
+[weakdeps]
+EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
+
+[extensions]
+EnzymeCoreExt = "EnzymeCore"
+
 [compat]
 Adapt = "4.0"
 julia = "1.6"
+EnzymeCore = "0.6, 0.7"

lib/GPUArraysCore/ext/EnzymeCoreExt.jl

@@ -0,0 +1,27 @@
+# compatibility with EnzymeCore
+
+module EnzymeCoreExt
+
+using GPUArraysCore
+
+if isdefined(Base, :get_extension)
+    using EnzymeCore
+    using EnzymeCore.EnzymeRules
+else
+    using ..EnzymeCore
+    using ..EnzymeCore.EnzymeRules
+end
+
+function EnzymeCore.EnzymeRules.inactive_noinl(::typeof(GPUArraysCore.default_scalar_indexing), args...)
+    return nothing
+end
+
+function EnzymeCore.EnzymeRules.inactive_noinl(::typeof(GPUArraysCore.assertscalar), args...)
+    return nothing
+end
+
+function EnzymeCore.EnzymeRules.inactive_noinl(::typeof(GPUArraysCore.allowscalar), args...)
+    return nothing
+end
+
+end # module

test/Project.toml

@@ -1,5 +1,6 @@
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
+EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
 JLArrays = "27aeb0d3-9eb9-45fb-866b-73c2ecf80fcb"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"

test/gpuarrayscore.jl

@@ -0,0 +1,9 @@
+using Test, GPUArraysCore, EnzymeCore
+
+@testset "EnzymeCore" begin
+    @test nothing == EnzymeCore.EnzymeRules.inactive_noinl(GPUArraysCore.assertscalar)
+
+    @test nothing == EnzymeCore.EnzymeRules.inactive_noinl(GPUArraysCore.default_scalar_indexing)
+
+    @test nothing == EnzymeCore.EnzymeRules.inactive_noinl(GPUArraysCore.allowscalar, identity)
+end

test/runtests.jl

@@ -1,5 +1,9 @@
 using GPUArrays, Test, Pkg
 
+@tests "GPUArraysCore" begin
+    include("gpuarrayscore.jl")
+end
+
 include("testsuite.jl")
 
 @testset "JLArray" begin