Disable Enzyme for AMDGPU

FluxML · Oct 2, 2023 · 3571306 · 3571306
1 parent 1b00dd1
commit 3571306
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -13,7 +13,9 @@ using Adapt
 using KernelAbstractions
 import ReverseDiff as RD        # used in `pooling.jl`
 
-const Test_Enzyme = VERSION <= v"1.10-" && !Sys.iswindows()
+const Test_Enzyme = VERSION <= v"1.10-" && !Sys.iswindows() &&
+    # TODO Enzyme is not working properly with AMDGPU yet.
+    get(ENV, "NNLIB_TEST_AMDGPU", "false") != "true"
 
 DocMeta.setdocmeta!(NNlib, :DocTestSetup, :(using NNlib, UnicodePlots); recursive=true)