diff --git a/test/fit.jl b/test/fit.jl index 78b672b..7612f52 100644 --- a/test/fit.jl +++ b/test/fit.jl @@ -694,9 +694,9 @@ end if CUDA.functional() push!(methods_vec, :CUDA) end - #if Metal.functional() - # push!(methods_vec, :Metal) - #end + if Metal.functional() + push!(methods_vec, :Metal) + end df = DataFrame(CSV.File(joinpath(dirname(pathof(FixedEffectModels)), "../dataset/EmplUK.csv"))) for method in methods_vec # same thing with float32 precision diff --git a/test/predict.jl b/test/predict.jl index f1a85f7..102e4d3 100644 --- a/test/predict.jl +++ b/test/predict.jl @@ -399,9 +399,9 @@ end if CUDA.functional() push!(methods_vec, :CUDA) end - #if Metal.functional() - # push!(methods_vec, :Metal) - #end + if Metal.functional() + push!(methods_vec, :Metal) + end for method in methods_vec local model = @formula Sales ~ Price + fe(Year) local result = reg(df, model, save = true, method = method, double_precision = false) diff --git a/test/runtests.jl b/test/runtests.jl index eebbe16..523bfb0 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -1,4 +1,4 @@ -using FixedEffectModels, Test +using CUDA, Metal, FixedEffectModels, Test @testset "formula" begin include("formula.jl") end @testset "fit" begin include("fit.jl") end