up (#271)

matthieugomez · web-flow · commit 0e06a544ec8c · 2026-01-21T15:16:39.000-06:00
diff --git 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
@@ -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
@@ -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

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-using FixedEffectModels, Test`
	`1`	`+using CUDA, Metal, FixedEffectModels, Test`
`2`	`2`
`3`	`3`	`@testset "formula" begin include("formula.jl") end`
`4`	`4`	`@testset "fit" begin include("fit.jl") end`