From 66910bf560376d8c02c6e4ef91941e2d92516384 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= <765740+giordano@users.noreply.github.com> Date: Tue, 21 Jan 2025 16:28:45 +0100 Subject: [PATCH] Fix condition to skip CUDA tests on aarch64 (#592) * Fix condition to skip CUDA tests on aarch64 * Skip more tests --- test/integration/cuda.jl | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/test/integration/cuda.jl b/test/integration/cuda.jl index 8cb3475bb..e1235d053 100644 --- a/test/integration/cuda.jl +++ b/test/integration/cuda.jl @@ -19,8 +19,8 @@ end # GPUCompiler throws "Not implemented" errors on aarch64 before # for some tests. const skip_tests = - Base.BinaryPlatforms.arch(Base.BinaryPlatforms.HostPlatform()) == "aarch" && - VERSION <= v"1.11.2" + Base.BinaryPlatforms.arch(Base.BinaryPlatforms.HostPlatform()) == "aarch64" && + VERSION <= v"1.11.3" @static if !Sys.isapple() @testset "Square Kernel" begin @@ -126,7 +126,9 @@ tuplef2(a) = @cuda threads = 1 tuplef2!((5, a)) @jit tuplef2(A) @test all(Array(A) .≈ 5) else - @code_hlo optimize = :before_kernel tuplef2(A) + @static if !skip_tests + @code_hlo optimize = :before_kernel tuplef2(A) + end end end A = ConcreteRArray(fill(1)) @@ -162,7 +164,9 @@ end @jit aliased(a) @test all(Array(a) .== 9) else - @code_hlo optimize = :before_kernel aliased(a) + @static if !skip_tests + @code_hlo optimize = :before_kernel aliased(a) + end end end end