[user-streams] Track symbolic current stream

[mlazos]

Stack from ghstack (oldest at bottom):

• [dynamo] Remove retrieving objects by ID #162905
• [user-streams] Add basic stream tests #164523
• [user-streams] Handle returning the current stream with/without device index #165356
• -> [user-streams] Track symbolic current stream #165212
• [user-streams] Add current stream source #165211
• [user-streams] Fix stream graph output semantics #164819
• [User-streams] Make torch.Event weakref compatible #164522
• [user-streams] Make device-agnostic streams weakref compatible #164304
• [user-cuda-streams] Add cuda streams test suite #162901
• [user-streams] Support streams as contexts #164507
• [user-streams] Have StreamVariable inherit from StreamContextVariable #164344
• [user-streams] Move StreamContextVariable into streams module #164343
• [user-streams] update stream context to use fork/join #162903
• [user-cuda-streams] Add fork/join custom ops #162900
• [user-streams] Handle aliasing properly #163028
• [user-cuda-streams] Pass streams/events to the graph via lookup table #162899
• [user-streams] Move stream code to streams module #163027

merge into stream tests

[pytorch-bot]

🔗 Helpful Links

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

• 📄 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

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

❗ 2 Active SEVs

There are 2 currently active SEVs. If your PR is affected, please view them below:

• Macos CI runners unavailable
• [ROCm][CI] Machines under the label linux.rocm.gpu.2 are undergoing maintenance.

❌ 56 New Failures

As of commit 06400d7 with merge base ecb5307 ():

NEW FAILURES - The following jobs have failed:

• Check Labels / Check labels (gh)
  RuntimeError: Error checking labels: PR does not have required labels
• Check mergeability of ghstack PR / ghstack-mergeability-check (gh)
  RuntimeError: Found an unexpected number of PRs mentioned in commit 66c8640559b4269c399e3383ad119806965222cc: 2. This is probably because you are using an old version of ghstack. Please update ghstack and resubmit your PRs
• inductor / inductor-cpu-test / test (cpu_inductor_torchbench, 1, 2, linux.8xlarge.amx) (gh)
  lennard_jones
• inductor / inductor-cpu-test / test (cpu_inductor_torchbench, 2, 2, linux.8xlarge.amx) (gh)
  vgg16
• inductor / inductor-cpu-test / test (dynamic_cpu_inductor_huggingface, 1, 1, linux.8xlarge.amx) (gh)
  YituTechConvBert
• inductor / inductor-cpu-test / test (dynamic_cpu_inductor_timm, 1, 2, linux.8xlarge.amx) (gh)
  mobilenetv2_100
• inductor / inductor-cpu-test / test (dynamic_cpu_inductor_timm, 2, 2, linux.8xlarge.amx) (gh)
  visformer_small
• inductor / inductor-cpu-test / test (dynamic_cpu_inductor_torchbench, 1, 2, linux.8xlarge.amx) (gh)
  lennard_jones
• inductor / inductor-cpu-test / test (dynamic_cpu_inductor_torchbench, 2, 2, linux.8xlarge.amx) (gh)
  vgg16
• inductor / inductor-cpu-test / test (inductor_torchbench_cpu_smoketest_perf, 1, 1, linux.24xl.spr-metal) (gh)
  Process completed with exit code 255.
• inductor / unit-test / inductor-cpu-test / test (inductor_amx, 1, 2, linux.8xlarge.amx) (gh)
  Process completed with exit code 1.
• inductor / unit-test / inductor-cpu-test / test (inductor_amx, 2, 2, linux.8xlarge.amx) (gh)
  Process completed with exit code 1.
• inductor / unit-test / inductor-cpu-test / test (inductor_avx2, 1, 2, linux.10xlarge.avx2) (gh)
  Process completed with exit code 1.
• inductor / unit-test / inductor-cpu-test / test (inductor_avx2, 2, 2, linux.10xlarge.avx2) (gh)
  Process completed with exit code 1.
• inductor / unit-test / inductor-halide-test / test (inductor-halide, 1, 1, linux.12xlarge) (gh)
  test/inductor/test_halide.py::HalideCpuTests::test_large_block_sizes_cpu_halide
• inductor / unit-test / linux-jammy-cpu-py3.12-gcc11-inductor-triton-cpu / test (inductor-triton-cpu, 1, 1, linux.12xlarge) (gh)
  test/inductor/test_triton_cpu_backend.py::CpuTritonTests::test_large_block_sizes_cpu
• pull / linux-docs / build-docs-python-false (gh)
  RuntimeError: no running event loop
• pull / linux-jammy-py3.10-clang12 / test (crossref, 1, 2, lf.linux.2xlarge) (gh)
  test/test_fake_tensor.py::FakeTensorTest::test_export_numpy
• pull / linux-jammy-py3.10-clang12 / test (crossref, 2, 2, lf.linux.2xlarge) (gh)
  test/export/test_export.py::TestDynamismExpression::test_export_strict_narrow_unbacked_expr
• pull / linux-jammy-py3.10-clang12 / test (default, 1, 5, lf.linux.4xlarge) (gh)
  test/inductor/test_flex_attention.py::TestFlexAttentionCPU::test_GQA_causal_mask_cpu
• pull / linux-jammy-py3.10-clang12 / test (default, 2, 5, lf.linux.4xlarge) (gh)
  test/test_fake_tensor.py::FakeTensorTest::test_export_numpy
• pull / linux-jammy-py3.10-clang12 / test (default, 3, 5, lf.linux.4xlarge) (gh)
  test/inductor/test_torchinductor.py::CpuTests::test_large_block_sizes_cpu
• pull / linux-jammy-py3.10-clang12 / test (default, 4, 5, lf.linux.4xlarge) (gh)
  test/dynamo/test_repros.py::ReproTests::test_311_resume_block_keyerror
• pull / linux-jammy-py3.10-clang12 / test (default, 5, 5, lf.linux.4xlarge) (gh)
  test/dynamo/test_dynamic_shapes.py::DynamicShapesReproTests::test_dont_dce_rand_dynamic_shapes
• pull / linux-jammy-py3.10-clang12 / test (dynamo_wrapped, 1, 3, lf.linux.2xlarge) (gh)
  Process completed with exit code 1.
• pull / linux-jammy-py3.10-clang12 / test (dynamo_wrapped, 2, 3, lf.linux.2xlarge) (gh)
  Process completed with exit code 1.
• pull / linux-jammy-py3.10-clang12 / test (dynamo_wrapped, 3, 3, lf.linux.2xlarge) (gh)
  Process completed with exit code 1.
• pull / linux-jammy-py3.10-clang12 / test (einops, 1, 1, lf.linux.2xlarge) (gh)
  test/dynamo/test_einops.py::TestEinops::test_layers_version_0_6_1
• pull / linux-jammy-py3.10-clang12-onnx / test (default, 2, 2, lf.linux.2xlarge) (gh)
  test/onnx/exporter/test_capture_strategies.py::ExportStrategiesTest::test_jit_isinstance_TorchExportStrictStrategy
• pull / linux-jammy-py3.10-clang18-asan / test (default, 1, 7, lf.linux.4xlarge) (gh)
  test/test_fx.py::TestDCE::test_impure_random
• pull / linux-jammy-py3.10-clang18-asan / test (default, 2, 7, lf.linux.4xlarge) (gh)
  test/test_fake_tensor.py::FakeTensorTest::test_export_numpy
• pull / linux-jammy-py3.10-clang18-asan / test (default, 3, 7, lf.linux.4xlarge) (gh)
  test/dynamo/test_dynamic_shapes.py::DynamicShapesReproTests::test_dont_dce_rand_dynamic_shapes
• pull / linux-jammy-py3.10-clang18-asan / test (default, 4, 7, lf.linux.4xlarge) (gh)
  test/dynamo/test_repros.py::ReproTests::test_311_resume_block_keyerror
• pull / linux-jammy-py3.10-clang18-asan / test (default, 5, 7, lf.linux.4xlarge) (gh)
  test/dynamo/test_dynamic_shapes.py::DynamicShapesCtxManagerTests::test_autocast_cpu_graph_break_2_dynamic_shapes
• pull / linux-jammy-py3.10-clang18-asan / test (default, 6, 7, lf.linux.4xlarge) (gh)
  test/inductor/test_torchinductor_codegen_dynamic_shapes.py::DynamicShapesCodegenCpuTests::test__dyn_quant_pack_4bit_weight_dynamic_shapes_cpu
• pull / linux-jammy-py3.10-clang18-asan / test (default, 7, 7, lf.linux.4xlarge) (gh)
  test/dynamo/test_higher_order_ops.py::HigherOrderOpTests::test_access_module_attr
• pull / linux-jammy-py3.10-gcc11 / test (default, 1, 5, lf.linux.2xlarge) (gh)
  test/inductor/test_flex_attention.py::TestFlexAttentionCPU::test_GQA_causal_mask_cpu
• pull / linux-jammy-py3.10-gcc11 / test (default, 2, 5, lf.linux.2xlarge) (gh)
  test/test_fake_tensor.py::FakeTensorTest::test_export_numpy
• pull / linux-jammy-py3.10-gcc11 / test (default, 3, 5, lf.linux.2xlarge) (gh)
  test/inductor/test_flex_decoding.py::TestFlexDecodingCPU::test_non_pow_2_headdim_head_dim_17_float32_cpu_float32
• pull / linux-jammy-py3.10-gcc11 / test (default, 4, 5, lf.linux.2xlarge) (gh)
  test/dynamo/test_repros.py::ReproTests::test_311_resume_block_keyerror
• pull / linux-jammy-py3.10-gcc11 / test (default, 5, 5, lf.linux.2xlarge) (gh)
  test/dynamo/test_dynamic_shapes.py::DynamicShapesReproTests::test_dont_dce_rand_dynamic_shapes
• pull / linux-jammy-py3.10-gcc11 / test (distributed, 1, 2, lf.linux.2xlarge) (gh)
  test/distributed/test_dynamo_distributed.py::TestFakeDistributedSingleProc::test_call_method_forward
• pull / linux-jammy-py3.10-gcc11 / test (distributed, 2, 2, lf.linux.2xlarge) (gh)
  test/distributed/tensor/test_dtensor_compile.py::TestDTensorCompile::test_device_mesh_compile
• pull / linux-jammy-py3.10-gcc11 / test (docs_test, 1, 1, lf.linux.2xlarge) (gh)
  RuntimeError: no running event loop
• pull / linux-jammy-py3.10-gcc11 / test (numpy_2_x, 1, 1, lf.linux.2xlarge) (gh)
  test/dynamo/test_unspec.py::UnspecTests::test_argmin_coerces_symint_to_intlist_spec
• pull / linux-jammy-py3.13-clang12 / test (crossref, 1, 2, lf.linux.2xlarge) (gh)
  test/test_fake_tensor.py::FakeTensorTest::test_export_numpy
• pull / linux-jammy-py3.13-clang12 / test (crossref, 2, 2, lf.linux.2xlarge) (gh)
  test/export/test_export.py::TestDynamismExpression::test_export_strict_narrow_unbacked_expr
• pull / linux-jammy-py3.13-clang12 / test (default, 1, 5, lf.linux.4xlarge) (gh)
  test/inductor/test_flex_attention.py::TestFlexAttentionCPU::test_GQA_causal_mask_cpu
• pull / linux-jammy-py3.13-clang12 / test (default, 2, 5, lf.linux.4xlarge) (gh)
  test/test_fake_tensor.py::FakeTensorTest::test_export_numpy
• pull / linux-jammy-py3.13-clang12 / test (default, 3, 5, lf.linux.4xlarge) (gh)
  test/inductor/test_torchinductor.py::CpuTests::test_large_block_sizes_cpu
• pull / linux-jammy-py3.13-clang12 / test (default, 4, 5, lf.linux.4xlarge) (gh)
  test/dynamo/test_dynamic_shapes.py::DynamicShapesReproTests::test_dont_dce_rand_dynamic_shapes
• pull / linux-jammy-py3.13-clang12 / test (default, 5, 5, lf.linux.4xlarge) (gh)
  test/inductor/test_torchinductor_dynamic_shapes.py::DynamicShapesCpuTests::test_large_block_sizes_dynamic_shapes_cpu
• pull / linux-jammy-py3.13-clang12 / test (dynamo_wrapped, 1, 3, lf.linux.2xlarge) (gh)
  test/test_torch.py::TestBasicVitalSigns::test_basic_vitals_read_write
• pull / linux-jammy-py3.13-clang12 / test (dynamo_wrapped, 2, 3, lf.linux.2xlarge) (gh)
  test/test_torch.py::TestBasicVitalSigns::test_basic_vitals
• pull / linux-jammy-py3.13-clang12 / test (dynamo_wrapped, 3, 3, lf.linux.2xlarge) (gh)
  test/test_nestedtensor.py::TestNestedTensorSubclassCPU::test_linear_backward_memory_usage_cpu_float32
• pull / linux-jammy-py3.13-clang12 / test (einops, 1, 1, lf.linux.2xlarge) (gh)
  test/dynamo/test_einops.py::TestEinops::test_layers_version_0_6_1

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

[github-actions]

This PR needs a release notes: label

If your changes are user facing and intended to be a part of release notes, please use a label starting with release notes:.

If not, please add the topic: not user facing label.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "topic: not user facing"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.