Merge pull request #828

Add Julia 1.11 Enzyme job to pipeline

Author: luraess

.buildkite/pipeline.yml

@@ -111,7 +111,25 @@ steps:
       JULIA_AMDGPU_CORE_MUST_LOAD: "1"
       JULIA_AMDGPU_HIP_MUST_LOAD: "1"
       JULIA_AMDGPU_DISABLE_ARTIFACTS: "1"
-    soft_fail: true
+
+  - label: "Julia 1.11 Enzyme"
+    plugins:
+      - JuliaCI/julia#v1:
+          version: "1.11"
+      - JuliaCI/julia-test#v1:
+          test_args: "enzyme"
+    agents:
+      queue: "juliagpu"
+      rocm: "*"
+      rocmgpu: "*"
+    if: build.message !~ /\[skip tests\]/
+    command: "julia --project -e 'using Pkg; Pkg.update()'"
+    timeout_in_minutes: 30
+    env:
+      JULIA_NUM_THREADS: 4
+      JULIA_AMDGPU_CORE_MUST_LOAD: "1"
+      JULIA_AMDGPU_HIP_MUST_LOAD: "1"
+      JULIA_AMDGPU_DISABLE_ARTIFACTS: "1"
 
   - label: "GPU-less environment"
     plugins:

ext/AMDGPUEnzymeCoreExt/AMDGPUEnzymeCoreExt.jl

@@ -14,6 +14,14 @@ function EnzymeCore.compiler_job_from_backend(
     return GPUCompiler.CompilerJob(mi, AMDGPU.compiler_config(AMDGPU.device()))
 end
 
+function EnzymeRules.forward(
+    config, fn::Const{typeof(AMDGPU.hipfunction)}, ::Type{<: Const},
+    f::Const{F}, tt::Const{TT}; kwargs...,
+) where {F, TT}
+    res = fn.val(f.val, tt.val; kwargs...)
+    return res
+end
+
 function EnzymeRules.forward(
     config, fn::Const{typeof(AMDGPU.hipfunction)}, ::Type{<: Duplicated},
     f::Const{F}, tt::Const{TT}; kwargs...,