fix: remove selectifactive

avik-pal · avik-pal · commit fd833419012a · 2025-11-02T15:41:15.000-06:00
diff --git a/src/enzyme_ad/jax/Implementations/HLODerivatives.td b/src/enzyme_ad/jax/Implementations/HLODerivatives.td
@@ -1559,12 +1559,5 @@ def : HLODerivative<"BroadcastInDimOp", (Op $x),
         )
       ],
       (
-        SelectIfActive $x,
-        (
-          BroadcastInDim
-          (ResultTypeWithBatch),
-          (Shadow $x),
-          (getBroadcastDimensionsWithBatch)
-        ),
-        (HLOConstantFP<"0">)
+        BroadcastInDim (ResultTypeWithBatch), (Shadow $x), (getBroadcastDimensionsWithBatch)
       )>;

Original file line number	Diff line number	Diff line change
`@@ -1559,12 +1559,5 @@ def : HLODerivative<"BroadcastInDimOp", (Op $x),`
`1559`	`1559`	`)`
`1560`	`1560`	`],`
`1561`	`1561`	`(`
`1562`		`- SelectIfActive $x,`
`1563`		`- (`
`1564`		`- BroadcastInDim`
`1565`		`- (ResultTypeWithBatch),`
`1566`		`- (Shadow $x),`
`1567`		`- (getBroadcastDimensionsWithBatch)`
`1568`		`- ),`
`1569`		`- (HLOConstantFP<"0">)`
	`1562`	`+ BroadcastInDim (ResultTypeWithBatch), (Shadow $x), (getBroadcastDimensionsWithBatch)`
`1570`	`1563`	`)>;`