chore: testing Enzyme 0.13 (#119)

* chore: testing Enzyme 0.13

* ci: preserve names

Author: avik-pal

.buildkite/pipeline.yml

@@ -3,7 +3,6 @@ steps:
     matrix:
       setup:
         version:
-          - "1.9"
           - "1.10"
     plugins:
       - JuliaCI/julia#v1:

.github/workflows/CI.yml

@@ -21,9 +21,8 @@ jobs:
       fail-fast: false
       matrix:
         version:
-          - '1.9'
           - '1.10'
-          - ~1.11.0-0
+          - '~1.11.0-0'
           - 'nightly'
         os:
           - ubuntu-20.04
@@ -34,11 +33,6 @@ jobs:
           - false
         libReactant: [packaged]
         include:
-          - os: ubuntu-20.04
-            arch: x64
-            libReactant: packaged
-            version: '1.9'
-            assertions: true
           - os: ubuntu-20.04
             arch: x64
             libReactant: packaged

Project.toml

@@ -29,13 +29,13 @@ Adapt = "4"
 ArrayInterface = "7.10"
 CEnum = "0.4, 0.5"
 Downloads = "1.6"
-Enzyme = "0.11, 0.12"
+Enzyme = "0.13"
 NNlib = "0.9"
 Preferences = "1.4"
 Reactant_jll = "0.0.19"
 Scratch = "1.2"
-Statistics = "1.9"
-julia = "1.9"
+Statistics = "1.10"
+julia = "1.10"
 
 [extras]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/utils.jl

@@ -92,24 +92,6 @@ function make_mlir_fn(f, args, kwargs, name="main", concretein=true; toscalar=fa
             Base.code_ircode(f, map(typeof, traced_args); interp),
         ))
 
-        # NOTE on Julia 1.9, it appends a ghost argument at the end
-        # solution: manually specify argument types
-        @static if VERSION < v"1.10"
-            empty!(ir.argtypes)
-            if f === Reactant.apply
-                append!(
-                    ir.argtypes,
-                    Any[
-                        Core.Const(f),
-                        typeof(traced_args[1]),
-                        Tuple{typeof.(traced_args[2:end])...},
-                    ],
-                )
-            else
-                append!(ir.argtypes, Any[Core.Const(f), typeof.(traced_args)...])
-            end
-        end
-
         oc = Core.OpaqueClosure(ir)
 
         if f === Reactant.apply

test/runtests.jl

@@ -48,7 +48,4 @@ include("closure.jl")
 include("compile.jl")
 include("buffer_donation.jl")
 include("nn.jl")
-
-if VERSION ≥ v"1.10-" # Lux isn't supported on 1.9
-    include("nn_lux.jl")
-end
+include("nn_lux.jl")