pytorch - 💡(How to fix) Fix DISABLED test_dtensor_basic_compile (__main__.TestDTensorCompileWithCompiledAutograd) [1 comments, 2 participants]

Fix Plan

The fix involves re-enabling and updating the test case test_dtensor_basic_compile in test_compiled_autograd.py to ensure compatibility with the main branch.

Steps to Fix

• Re-enable the disabled test case by removing the decorator or comment that disables it.
• Update the test case to handle any changes in the DTensor or CompiledAutograd APIs.
• Verify that the test case passes on the main branch.

Example Code

# test_compiled_autograd.py
import unittest
from torch import dtensor

class TestDTensorCompileWithCompiledAutograd(unittest.TestCase):
    def test_dtensor_basic_compile(self):
        # Update the test case to handle any API changes
        dtensor_tensor = dtensor.randn(2, 2, device="xpu")
        # Add assertions to verify the test case
        self.assertEqual(dtensor_tensor.device, "xpu")

Verification

Run the test case using the command python -m unittest test_compiled_autograd.py and verify that it passes.

Extra Tips

• Ensure that the xpu device is properly configured and available.
• Review the test case and update it to handle any edge cases or errors that may occur.