Fix BLAS on Windows (#260)

EnzymeAD · Mar 15, 2022 · e0ae6f9 · e0ae6f9 · vchuravy · Mar 15, 2022
1 parent 0742289
commit e0ae6f9
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/compiler/validation.jl b/src/compiler/validation.jl
@@ -35,15 +35,23 @@ module FFI
             function get_blas_symbols()
                 symbols = Set{String}()
                 path = Libdl.dlpath(BLAS.libblas)
-                ignoreSymbols = Set(String["", "edata", "_edata", "end", "_end", "_bss_start", "__bss_start"])
+                ignoreSymbols = Set(String["", "edata", "_edata", "end", "_end", "_bss_start", "__bss_start", ".text", ".data"])
                 for s in Symbols(readmeta(open(path, "r")))
                     name = symbol_name(s)
-                    BLAS.vendor() == :openblas64 && endswith(name, "64_") || continue
+                    if !Sys.iswindows() && BLAS.vendor() == :openblas64
+                        endswith(name, "64_") || continue
+                    else
+                        endswith(name, "_") || continue
+                    end
                     if !in(name, ignoreSymbols)
                         push!(symbols, name)
                     end
                 end
-                return collect(symbols)
+                symbols = collect(symbols)
+                if Sys.iswindows() &&  BLAS.vendor() == :openblas64
+                    return map(n->n*"64_", symbols)
+                end
+                return symbols
             end
 
             function lookup_blas_symbol(name)