fix flashinfer

Author: lanluo-nvidia

tests/py/dynamo/automatic_plugin/test_flashinfer_rmsnorm.py

@@ -12,12 +12,11 @@
 from ..conversion.harness import DispatchTestCase
 
 if importlib.util.find_spec("flashinfer"):
-    import flashinfer
+    # flashinfer has been impacted by torch upstream change: https://github.com/pytorch/pytorch/commit/660b0b8128181d11165176ea3f979fa899f24db1
+    # got ImportError: cannot import name '_get_pybind11_abi_build_flags' from 'torch.utils.cpp_extension'
+    # import flashinfer
 
 
-# flashinfer has been impacted by torch upstream change: https://github.com/pytorch/pytorch/commit/660b0b8128181d11165176ea3f979fa899f24db1
-# got ImportError: cannot import name '_get_pybind11_abi_build_flags' from 'torch.utils.cpp_extension'
-@unittest.skip("Not Available")
 @torch.library.custom_op("flashinfer::rmsnorm", mutates_args=())  # type: ignore[misc]
 def flashinfer_rmsnorm(
     input: torch.Tensor, weight: torch.Tensor, eps: float = 1e-6