Update xla_native_functions.yaml

pytorch · Jan 17, 2024 · 7837b95 · 7837b95
1 parent f7a51bf
commit 7837b95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/codegen/xla_native_functions.yaml b/codegen/xla_native_functions.yaml
@@ -66,7 +66,6 @@ full_codegen:
   - masked_fill.Scalar
   - masked_fill.Tensor
   - maximum
-  - map
   - minimum
   - native_dropout_backward
   - ne.Scalar
@@ -222,6 +221,7 @@ supported:
   - log2
   - log10
   - logsumexp
+  - map
   - masked_scatter
   - masked_select
   - max