diff --git a/functorch/csrc/BatchRulesUnaryOps.cpp b/functorch/csrc/BatchRulesUnaryOps.cpp index 935514179..a42c92e0f 100644 --- a/functorch/csrc/BatchRulesUnaryOps.cpp +++ b/functorch/csrc/BatchRulesUnaryOps.cpp @@ -41,8 +41,10 @@ TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) { UNARY_POINTWISE(glu); UNARY_POINTWISE(isfinite); UNARY_POINTWISE(isnan); + UNARY_POINTWISE(isinf); UNARY_POINTWISE(isposinf); UNARY_POINTWISE(isneginf); + UNARY_POINTWISE(isreal); UNARY_POINTWISE_ALL(lgamma); UNARY_POINTWISE_ALL(log); UNARY_POINTWISE_ALL(log10); diff --git a/test/test_vmap.py b/test/test_vmap.py index 12a23a841..722a91a7a 100644 --- a/test/test_vmap.py +++ b/test/test_vmap.py @@ -3173,8 +3173,6 @@ def test_vmap_exhaustive(self, device, dtype, op): xfail('fft.ihfftn'), xfail('fft.irfft2'), xfail('fft.rfft2'), - xfail('isinf'), - xfail('isreal'), xfail('allclose'), xfail('argwhere'), xfail('bfloat16', 'channels_last'),