test: syrk

Author: avik-pal

benchmark/misc/newton_schulz.jl

@@ -35,7 +35,7 @@ function run_newton_schulz_benchmark!(results, backend)
     compile_modes = [
         (
             "StructuredTensors",
-            Reactant.CompileOptions(; sync=true, disable_structured_tensors_passes=false)
+            Reactant.CompileOptions(; sync=true, disable_structured_tensors_passes=false),
         ),
         (
             "Default",

test/integration/linear_algebra.jl

@@ -737,10 +737,16 @@ raise_to_syrk2(x, y) = 3 .* (transpose(x) * x) .+ 5 .* y
         @testset for (fn, y) in ((raise_to_syrk, y1), (raise_to_syrk2, y2))
             y_ra = Reactant.to_rarray(y)
 
-            hlo = @code_hlo optimize = :before_jit fn(x_ra, y_ra)
+            hlo = @code_hlo compile_options = CompileOptions(;
+                disable_structured_tensors_passes=false, optimize=:before_jit
+            ) fn(x_ra, y_ra)
             @test occursin("enzymexla.blas.syrk", repr(hlo))
 
-            @test @jit(fn(x_ra, y_ra)) ≈ fn(x, y)
+            fn_compile = @compile compile_options = CompileOptions(;
+                disable_structured_tensors_passes=false
+            ) fn(x_ra, y_ra)
+
+            @test fn_compile(x_ra, y_ra) ≈ fn(x, y)
         end
     end
 end