From f14ad34dfde2323d50b0339bec2e88a2bb729aee Mon Sep 17 00:00:00 2001 From: William Moses Date: Sun, 29 Sep 2024 13:21:51 -0500 Subject: [PATCH] Bump jll (#1922) --- Project.toml | 4 ++-- test/runtests.jl | 27 +++++++++++++++++++++++++++ 2 files changed, 29 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index 21d0cab215..87ab945958 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "Enzyme" uuid = "7da242da-08ed-463a-9acd-ee780be4f1d9" authors = ["William Moses ", "Valentin Churavy "] -version = "0.13.4" +version = "0.13.5" [deps] CEnum = "fa961155-64e5-5f13-b03f-caf6b980ea82" @@ -36,7 +36,7 @@ BFloat16s = "0.2, 0.3, 0.4, 0.5" CEnum = "0.4, 0.5" ChainRulesCore = "1" EnzymeCore = "0.8.4" -Enzyme_jll = "0.0.152" +Enzyme_jll = "0.0.153" GPUCompiler = "0.21, 0.22, 0.23, 0.24, 0.25, 0.26, 0.27" LLVM = "6.1, 7, 8, 9" LogExpFunctions = "0.3" diff --git a/test/runtests.jl b/test/runtests.jl index ba4462bc23..902b9e4f65 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -3585,6 +3585,33 @@ end @test res[3].offset1 == 0 @test res[2].stride1 == 1 @test res[3].stride1 == 1 + + x = [Float32(0.25)] + dx = [Float32(0.0)] + rng = Base.UnitRange{Int64}(1, 0) + + f = Const(Base.SubArray{T, N, P, I, L} where L where I where P where N where T) + a1 = Const(Base.IndexLinear()) + a2 = Duplicated(x, dx) + a3 = Const((rng,)) + a4 = Const((true,)) + + fwd, rev = autodiff_thunk(set_runtime_activity(ReverseSplitWithPrimal), + typeof(f), + Duplicated, + typeof(a1), + typeof(a2), + typeof(a3), + typeof(a4) + ) + + res = fwd(f,a1,a2,a3,a4) + @test res[2].indices == (rng,) + @test res[3].indices == (rng,) + @test res[2].offset1 == 0 + @test res[3].offset1 == 0 + @test res[2].stride1 == 1 + @test res[3].stride1 == 1 end @testset "Uncached batch sizes" begin