code: def testDebugNansPjitWithDonation(self):
      if jax.device_count() < 2:
          raise SkipTest("test requires >=2 devices")
  
      p = jax.sharding.PartitionSpec("x")
      f = pjit.pjit(
          lambda x: 0.0 / x,
          in_shardings=p,
          out_shardings=p,
          donate_argnums=(0,),
      )
  
      with jax.sharding.Mesh(np.array(jax.local_devices()[:2]), ("x",)):
          with self.assertRaises(FloatingPointError):
              ans = f(jnp.array([0.0, 1.0]))
              ans.block_until_ready()
  
imports: from absl.testing import absltest
  import jax
  import numpy
  from unittest import SkipTest
  from jax._src import api
  from jax._src import test_util
  from jax import numpy
  from jax.experimental import pjit, maps
  from jax import config
label_desc: Use image node_id 17 for calling the DebugNaNsTest obj's underlying member method code with example usage: obj.testDebugNansPjitWithDonation() without return types