test(distributed): compare DTensor basic ops against eager, not torch.compile [14/n]

[bhimrazy]

What does this PR do?

Compares test_dtensor_basic_op against eager instead of torch.compile, as the opinfo tests in the same file already do.

Problem: the test built its reference with torch.compile, so an inductor Triton launch ran before thunder was reached. On torch nightly that launch fails with CUDA driver error: invalid argument, and the other rank then waits for the 300s process timeout. So one cause shows up as a crash in one variant and as timeouts in two others. The test is about thunder's lowering of the op over a DTensor, not inductor's.

Checked on 2× L4 with the pt_main-dev image: 3 of the 6 variants fail before (11 min, with the timeouts), and all 6 pass after (55s). Binding each rank to its device first doesn't change the result.

Related CI:

• all-tests.yaml distributed on pt_main-dev: the three test_dtensor_basic_op_* failures.

CI log (distributed, pt_main-dev, on #2847)

  File ".../torch/_inductor/runtime/static_triton_launcher.py", line 291, in run
    self.C_impl._launch_kernel(
RuntimeError: CUDA driver error: invalid argument

FAILED thunder/tests/distributed/test_dtensor.py::DTensorTest::test_dtensor_basic_op_executor_torch_fn_key_x * w - RuntimeError: Process 1 exited with error code 10 and exception:
FAILED thunder/tests/distributed/test_dtensor.py::DTensorTest::test_dtensor_basic_op_executor_nvfuser_fn_key_x_mul(w) - RuntimeError: Process 0 terminated or timed out after 300.01899886131287 seconds
FAILED thunder/tests/distributed/test_dtensor.py::DTensorTest::test_dtensor_basic_op_executor_torch_fn_key_torch_mul - RuntimeError: Process 0 terminated or timed out after 300.0474462509155 seconds

Part 14 of the breakdown of #2832.