[CUDA] Implement MaskedScatter (#3151)

This commit is contained in:
Long Yixing
2026-03-15 10:33:55 +09:00
committed by GitHub
parent 5d1700493a
commit 0bdbfdb838
11 changed files with 289 additions and 53 deletions
-4
View File
@@ -39,8 +39,4 @@ 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",
}