[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
@@ -1357,11 +1357,6 @@ TEST_CASE("test grad dynamic slices") {
 }

 TEST_CASE("test masked_scatter autograd") {
-  if (cu::is_available()) {
-    INFO("Skipping masked_scatter cuda autograd tests");
-    return;
-  }
-
  // Test jvp
  {
    auto self = array({10.f, 20.f, 30.f, 40.f}, {4});