chore: mark EnzymeCore.augmented_primal as codecov excluded (#1467)

* chore: mark EnzymeCore.augmented_primal as codecov excluded

* ci: codecov status reporting

Author: avik-pal

.codecov.yml

@@ -0,0 +1,4 @@
+codecov:
+  require_ci_to_pass: false
+  notify:
+    wait_for_ci: false

ext/LuxLossFunctionsExt.jl

@@ -44,6 +44,8 @@ function CRC.rrule(
     return LossFunctionImpl.fused_agg(sum, lfn, x, y), ∇fused_agg
 end
 
+# COV_EXCL_START
+
 function EnzymeRules.augmented_primal(
     cfg::EnzymeRules.RevConfigWidth{1},
     func::EnzymeCore.Const{typeof(LossFunctionImpl.fused_agg)},
@@ -82,4 +84,6 @@ function EnzymeRules.reverse(
     return ntuple(Returns(nothing), 4)
 end
 
+# COV_EXCL_STOP
+
 end

lib/LuxLib/src/impl/batched_mul.jl

@@ -338,6 +338,8 @@ function CRC.rrule(
     return batched_matmul_fallback(x, y), ∇batched_matmul
 end
 
+# COV_EXCL_START
+
 # This is type-piracy but needed to fix a blocking issue. TODO: upstream to NNlib
 # Enzyme causes a "active variables passed by value to jl_new_task are not yet supported"
 # warning without this patch.
@@ -452,3 +454,5 @@ for func in (NNlib.batched_mul!, batched_matmul_loopvec_impl!)
         end
     end
 end
+
+# COV_EXCL_STOP

lib/LuxLib/src/impl/dense.jl

@@ -133,6 +133,8 @@ function CRC.rrule(
     return z, ∇fused_dense
 end
 
+# COV_EXCL_START
+
 # TODO: We can optimize these a bit further by checking for cases where the forward pass
 #       is not needed. We skip such optimizations for now
 function EnzymeRules.augmented_primal(
@@ -283,5 +285,7 @@ function EnzymeRules.reverse(
     return ntuple(Returns(nothing), 6)
 end
 
+# COV_EXCL_STOP
+
 ∇matmul_bias(∂y, weight, x, bias) = ∇matmul_bias(∂y, ∇bias_add(bias, ∂y), weight, x, bias)
 ∇matmul_bias(∂y, ∂b, weight, x, _) = matmul(∂y, x'), matmul(weight', ∂y), ∂b

lib/LuxLib/src/impl/matmul.jl

@@ -244,6 +244,8 @@ function CRC.rrule(
     return matmuladd(A, B, bias), ∇matmuladd
 end
 
+# COV_EXCL_START
+
 # EnzymeRules
 function EnzymeRules.augmented_primal(
     cfg,
@@ -340,6 +342,8 @@ function EnzymeRules.reverse(
     return ntuple(Returns(nothing), 5)
 end
 
+# COV_EXCL_STOP
+
 @enzyme_alternative matmul_octavian! matmul_linalg_default!
 @enzyme_alternative serial_matmul_loopvec! matmul_linalg_default!
 @enzyme_alternative matmul_loopvec! matmul_linalg_default!