diff --git a/test/forward_backward_compatibility/check_forward_backward_compatibility.py b/test/forward_backward_compatibility/check_forward_backward_compatibility.py index 9c27d9b791555..c14b36c7ad5af 100644 --- a/test/forward_backward_compatibility/check_forward_backward_compatibility.py +++ b/test/forward_backward_compatibility/check_forward_backward_compatibility.py @@ -139,11 +139,7 @@ ("aten::_weight_norm_cuda_interface", datetime.date(9999, 1, 1)), ("aten::_weight_norm_cuda_interface_backward", datetime.date(9999, 1, 1)), # TODO: FIXME: prims shouldn't be checked - ("prims::any", datetime.date(9999, 1, 1)), - ("prims::all", datetime.date(9999, 1, 1)), - ("prims::empty_like", datetime.date(9999, 1, 1)), - ("prims::select", datetime.date(9999, 1, 1)), - ("prims::concatenate", datetime.date(9999, 1, 1)), + ("prims::.*", datetime.date(9999, 1, 1)), ] ALLOW_LIST_COMPILED = [ diff --git a/torch/_prims/__init__.py b/torch/_prims/__init__.py index f962bf163007d..2d60da963e5bd 100644 --- a/torch/_prims/__init__.py +++ b/torch/_prims/__init__.py @@ -61,7 +61,7 @@ "exp2", "fill", "floor", - "is_finite", + "isfinite", "is_infinite", "lgamma", "log", @@ -626,8 +626,8 @@ def _isfinite_nvfuser(fd: Any, a: TensorLikeType): return fd.Ops.isfinite(a) # type: ignore[attr-defined] -is_finite = _make_elementwise_unary_prim( - "is_finite", +isfinite = _make_elementwise_unary_prim( + "isfinite", impl_aten=torch.isfinite, impl_nvfuser=_isfinite_nvfuser, doc="", diff --git a/torch/_refs/__init__.py b/torch/_refs/__init__.py index f2751eab2fb52..6cfb90ad49993 100644 --- a/torch/_refs/__init__.py +++ b/torch/_refs/__init__.py @@ -437,7 +437,7 @@ def _frac(x: TensorLikeType) -> TensorLikeType: def _isfinite(a: TensorLikeType) -> TensorLikeType: if utils.is_float_dtype(a.dtype) or utils.is_complex_dtype(a.dtype): - return prims.is_finite(a) + return prims.isfinite(a) return ones_like(a, dtype=torch.bool)