Add Masked Scatter (#2663)

Co-authored-by: Awni Hannun <[email protected]> Co-authored-by: Angelos Katharopoulos <[email protected]> Co-authored-by: Angelos Katharopoulos <[email protected]>
2025-11-19 14:53:32 -08:00
parent 7f4b7e553c
commit b3825ac149
26 changed files with 1099 additions and 51 deletions
@@ -56,4 +56,8 @@ cuda_skip = {
    "TestQuantized.test_throw",
    "TestQuantized.test_vjp_scales_biases",
    "TestExportImport.test_export_quantized_model",
+    # Masked scatter
+    "TestOps.test_masked_scatter",
+    "TestVmap.test_vmap_masked_scatter",
+    "TestArray.test_setitem_with_boolean_mask",
 }