typing scheduler.py [1/3]: Add NodeSchedulerNode

[aorenste]

Split off to make reviewing easier.

BaseSchedulerNode really represents two types of nodes - ones where the node field is valid (most of them) and ones where the node field isn't (ones derived from FusedSchedulerNode). In the FusedSchedulerNode case we also don't want to run BaseSchedulerNode's __init__. FusedSchedulerNode also overrides a bunch of methods to throw NotImplementedError.

Add a new class, NodeSchedulerNode to represent nodes where node is valid and move the methods that FusedSchedulerNode doesn't care about onto it.

No actual code changes - just moving methods around.

Stack from ghstack (oldest at bottom):

• typing scheduler.py [3/3]: update types #126505
• typing scheduler.py [1/2]: Bug fix #126610
• -> typing scheduler.py [1/3]: Add NodeSchedulerNode #126609

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/126609

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

Note: Links to docs will display an error until the docs builds have been completed.

✅ You can merge normally! (7 Unrelated Failures)

As of commit 95ea0a7 with merge base a44d0cf ():

FLAKY - The following job failed but was likely due to flakiness present on trunk:

• inductor / cuda12.1-py3.12-gcc9-sm86 / test (inductor, 1, 1, linux.g5.4xlarge.nvidia.gpu) (gh) (similar failure)
  inductor/test_aot_inductor.py::AOTInductorTestABICompatibleCuda::test_bmm_multiple_dynamic_abi_compatible_cuda

BROKEN TRUNK - The following jobs failed but was present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• inductor-periodic / cuda12.1-py3.10-gcc9-sm86-periodic-dynamo-benchmarks / test (aot_eager_torchbench, 1, 2, linux.g5.4xlarge.nvidia.gpu) (gh) (trunk failure)
  hf_BigBird
• inductor-periodic / cuda12.1-py3.10-gcc9-sm86-periodic-dynamo-benchmarks / test (dynamic_aot_eager_torchbench, 1, 2, linux.g5.4xlarge.nvidia.gpu) (gh) (trunk failure)
  hf_BigBird
• inductor-periodic / cuda12.1-py3.10-gcc9-sm86-periodic-dynamo-benchmarks / test (dynamo_eager_torchbench, 1, 2, linux.g5.4xlarge.nvidia.gpu) (gh) (trunk failure)
  hf_BigBird

UNSTABLE - The following jobs failed but were likely due to flakiness present on trunk and has been marked as unstable:

• inductor / cuda12.1-py3.10-gcc9-sm86 / test (aot_inductor_torchbench, 1, 2, linux.g5.4xlarge.nvidia.gpu, unstable) (gh) (#120841)
  hf_BigBird
• inductor / cuda12.1-py3.10-gcc9-sm86 / test (dynamic_inductor_torchbench, 1, 2, linux.g5.4xlarge.nvidia.gpu, unstable) (gh) (#120841)
  hf_BigBird
• inductor / cuda12.1-py3.10-gcc9-sm86 / test (inductor_torchbench, 1, 2, linux.g5.4xlarge.nvidia.gpu, unstable) (gh) (#120841)
  hf_BigBird

This comment was automatically generated by Dr. CI and updates every 15 minutes.