[CUDA] Implement MaskedScatter (#3151)

2026-03-15 10:33:55 +09:00
parent 5d1700493a
commit 0bdbfdb838
11 changed files with 289 additions and 53 deletions
@@ -2441,11 +2441,6 @@ TEST_CASE("test scatter") {
 }

 TEST_CASE("test masked_scatter") {
-  if (cu::is_available()) {
-    INFO("Skipping masked_scatter cuda ops tests");
-    return;
-  }
-
  // Wrong mask dtype
  CHECK_THROWS(masked_scatter(array({1, 2}), array({1, 2}), array({1, 2})));