[WIP][export][cond] support exporting cond with unbacked symint shaped tensor #154570

ydwu4 · May 28, 2025

Stack from ghstack (oldest at bottom):

-> [WIP][export][cond] support exporting cond with unbacked symint shaped tensor #154570

Still work in progerss.

[ghstack-poisoned]

pytorch-bot · May 28, 2025

🔗 Helpful Links

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

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

❌ 38 New Failures, 4 Unrelated Failures

As of commit 3536239 with merge base b6868f2 ():

NEW FAILURES - The following jobs have failed:

pull / linux-focal-cuda12.6-py3.10-gcc11 / test (default, 1, 5, linux.4xlarge.nvidia.gpu) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-focal-cuda12.6-py3.10-gcc11 / test (default, 2, 5, linux.4xlarge.nvidia.gpu) (gh)
'test/test_nestedtensor.py::TestNestedTensorOpInfoCUDA::test_compile_backward___rpow___cuda_float32'
pull / linux-focal-cuda12.6-py3.10-gcc11 / test (default, 3, 5, linux.4xlarge.nvidia.gpu) (gh)
dynamo/test_subclasses.py::SubclassTests::test_njt_subclass_from_buffer
pull / linux-focal-cuda12.6-py3.10-gcc11 / test (default, 4, 5, linux.4xlarge.nvidia.gpu) (gh)
functorch/test_aotdispatch.py::TestAOTAutograd::test_mark_activations_dynamic_with_nested
pull / linux-focal-cuda12.6-py3.10-gcc11 / test (default, 5, 5, linux.4xlarge.nvidia.gpu) (gh)
'test/test_nestedtensor.py::TestNestedTensorOpInfoCUDA::test_compile_backward___radd___cuda_float32'
pull / linux-focal-cuda12.6-py3.10-gcc11-sm89 / test (default, 1, 5, linux.g6.4xlarge.experimental.nvidia.gpu) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-focal-cuda12.6-py3.10-gcc11-sm89 / test (default, 2, 5, linux.g6.4xlarge.experimental.nvidia.gpu) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCUDA::test_compile_preserves_metadata_cache_cuda_float32
pull / linux-focal-cuda12.6-py3.10-gcc11-sm89 / test (default, 3, 5, linux.g6.4xlarge.experimental.nvidia.gpu) (gh)
dynamo/test_subclasses.py::SubclassTests::test_njt_subclass_from_buffer
pull / linux-focal-cuda12.6-py3.10-gcc11-sm89 / test (default, 4, 5, linux.g6.4xlarge.experimental.nvidia.gpu) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCUDA::test_compile_with_dynamic_min_seq_len_cuda_float32
pull / linux-focal-cuda12.6-py3.10-gcc11-sm89 / test (default, 5, 5, linux.g6.4xlarge.experimental.nvidia.gpu) (gh)
inductor/test_compiled_autograd.py::TestCompiledAutograd::test_free_activation_memory_subclass
pull / linux-focal-py3.13-clang10 / test (crossref, 1, 2, linux.2xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-focal-py3.13-clang10 / test (crossref, 2, 2, linux.2xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_to_padded_tensor_compile_nt_dim_2_requires_grad_True_cpu_float32
pull / linux-focal-py3.13-clang10 / test (default, 1, 5, linux.4xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-focal-py3.13-clang10 / test (default, 2, 5, linux.4xlarge) (gh)
dynamo/test_subclasses.py::SubclassTests::test_njt_subclass_from_buffer
pull / linux-focal-py3.13-clang10 / test (default, 3, 5, linux.4xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_to_padded_tensor_compile_nt_dim_2_requires_grad_True_cpu_float32
pull / linux-focal-py3.13-clang10 / test (default, 5, 5, linux.4xlarge) (gh)
functorch/test_aotdispatch.py::TestAOTAutogradWithDynamo::test_mark_activations_dynamic_with_nested
pull / linux-focal-py3.13-clang10 / test (dynamo_wrapped, 2, 3, linux.2xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-focal-py3.13-clang10 / test (dynamo_wrapped, 3, 3, linux.2xlarge) (gh)
test_nestedtensor.py::TestNestedTensorDeviceTypeCPU::test_embedding_jagged_cpu
pull / linux-focal-py3.9-clang10 / test (crossref, 1, 2, linux.2xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-focal-py3.9-clang10 / test (crossref, 2, 2, linux.2xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_compile_padded_dense_conversion_preserves_metadata_cache_cpu_float32
pull / linux-focal-py3.9-clang10 / test (default, 1, 5, linux.4xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-focal-py3.9-clang10 / test (default, 2, 5, linux.4xlarge) (gh)
functorch/test_aotdispatch.py::TestAOTAutogradWithDynamo::test_mark_activations_dynamic_with_nested
pull / linux-focal-py3.9-clang10 / test (default, 3, 5, linux.4xlarge) (gh)
functorch/test_aotdispatch.py::TestAOTAutograd::test_mark_activations_dynamic_with_nested
pull / linux-focal-py3.9-clang10 / test (default, 4, 5, linux.4xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_compile_padded_dense_conversion_preserves_metadata_cache_cpu_float32
pull / linux-focal-py3.9-clang10 / test (default, 5, 5, linux.4xlarge) (gh)
dynamo/test_subclasses.py::SubclassTests::test_njt_subclass_from_buffer
pull / linux-focal-py3.9-clang10 / test (dynamo_wrapped, 2, 3, linux.2xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-focal-py3.9-clang10 / test (dynamo_wrapped, 3, 3, linux.2xlarge) (gh)
test_nestedtensor.py::TestNestedTensorDeviceTypeCPU::test_embedding_jagged_cpu
pull / linux-jammy-py3.10-clang15-asan / test (default, 1, 6, linux.4xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-jammy-py3.10-clang15-asan / test (default, 2, 6, linux.4xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_to_padded_tensor_compile_nt_dim_2_requires_grad_True_cpu_float32
pull / linux-jammy-py3.10-clang15-asan / test (default, 3, 6, linux.4xlarge) (gh)
dynamo/test_subclasses.py::SubclassTests::test_njt_subclass_from_buffer
pull / linux-jammy-py3.10-clang15-asan / test (default, 4, 6, linux.4xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_to_padded_tensor_compile_nt_dim_2_requires_grad_False_cpu_float16
pull / linux-jammy-py3.10-clang15-asan / test (default, 5, 6, linux.4xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_compile_padded_dense_conversion_preserves_metadata_cache_cpu_float32
pull / linux-jammy-py3.10-clang15-asan / test (default, 6, 6, linux.4xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_compile_preserves_metadata_cache_cpu_float32
pull / linux-jammy-py3.9-gcc11 / test (default, 1, 5, linux.2xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved
pull / linux-jammy-py3.9-gcc11 / test (default, 2, 5, linux.2xlarge) (gh)
dynamo/test_subclasses.py::SubclassTests::test_njt_subclass_from_buffer
pull / linux-jammy-py3.9-gcc11 / test (default, 3, 5, linux.2xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_compile_with_propagated_dynamic_max_seq_len_cpu_float32
pull / linux-jammy-py3.9-gcc11 / test (default, 5, 5, linux.2xlarge) (gh)
test_nestedtensor.py::TestNestedTensorSubclassCPU::test_compile_padded_dense_conversion_preserves_metadata_cache_cpu_float32
pull / linux-jammy-py3.9-gcc11 / test (numpy_2_x, 1, 1, linux.2xlarge) (gh)
test_fake_tensor.py::FakeTensorTest::test_jagged_fake_to_fake_preserved

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

inductor / cuda12.8-py3.10-gcc9-sm86 / test (inductor_torchbench, 2, 2, linux.g5.4xlarge.nvidia.gpu) (gh) (similar failure)
opacus_cifar10
inductor / linux-jammy-cpu-py3.9-gcc11-inductor / test (cpu_inductor_torchbench, 2, 2, linux.8xlarge.amx) (gh) (similar failure)
opacus_cifar10
inductor / linux-jammy-cpu-py3.9-gcc11-inductor / test (dynamic_cpu_inductor_torchbench, 2, 2, linux.8xlarge.amx) (gh) (similar failure)
opacus_cifar10

BROKEN TRUNK - The following job failed but were present on the merge base:

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

pull / linux-jammy-py3-clang12-executorch / test (executorch, 1, 1, linux.2xlarge) (gh) (trunk failure)
Process completed with exit code 1.

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

ghstack-source-id: b312fec Pull Request resolved: #154570

ezyang · Jun 6, 2025

Design doc?

ydwu4 · Jun 6, 2025

Don't have bandwidth this week. Will go back to it next week. Probably need some more discussion.

[export][cond] support exporting cond with unbacked symint shaped tensor

3536239

[ghstack-poisoned]

ydwu4 requested review from bobrenjc93 and laithsakka as code owners May 28, 2025 22:25

ydwu4 added a commit that referenced this pull request May 28, 2025

[export][cond] support exporting cond with unbacked symint shaped tensor

7439925

ghstack-source-id: b312fec Pull Request resolved: #154570

pytorch-bot bot added ciflow/inductor release notes: fx release notes category labels May 28, 2025

facebook-github-bot added the fx label May 28, 2025

ydwu4 changed the title ~~[export][cond] support exporting cond with unbacked symint shaped tensor~~ [WIP][export][cond] support exporting cond with unbacked symint shaped tensor May 28, 2025

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

[WIP][export][cond] support exporting cond with unbacked symint shaped tensor #154570

[WIP][export][cond] support exporting cond with unbacked symint shaped tensor #154570

Uh oh!

ydwu4 commented May 28, 2025 •

edited

Loading

Uh oh!

pytorch-bot bot commented May 28, 2025 •

edited

Loading

Uh oh!

ezyang commented Jun 6, 2025

Uh oh!

ydwu4 commented Jun 6, 2025

Uh oh!

Uh oh!

Search code, repositories, users, issues, pull requests...

[WIP][export][cond] support exporting cond with unbacked symint shaped tensor #154570

Are you sure you want to change the base?

[WIP][export][cond] support exporting cond with unbacked symint shaped tensor #154570

Uh oh!

Conversation

ydwu4 commented May 28, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

pytorch-bot bot commented May 28, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

🔗 Helpful Links

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

❌ 38 New Failures, 4 Unrelated Failures

Uh oh!

ezyang commented Jun 6, 2025

Uh oh!

ydwu4 commented Jun 6, 2025

Uh oh!

Uh oh!

ydwu4 commented May 28, 2025 •

edited

Loading

pytorch-bot bot commented May 28, 2025 •

edited

Loading