Use assert_scalar to emit value range checks

[ezyang]

Stack from ghstack (oldest at bottom):

• -> Use assert_scalar to emit value range checks #126682

As we now have range refinement on relational bounds, this reduces the
target operator set.

Signed-off-by: Edward Z. Yang ezyang@meta.com

[pytorch-bot]

🔗 Helpful Links

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

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

❌ 21 New Failures, 2 Unrelated Failures

As of commit 6ef7b68 with merge base 2f53747 ():

NEW FAILURES - The following jobs have failed:

• pull / linux-focal-cuda12.1-py3.10-gcc9 / test (default, 1, 5, linux.4xlarge.nvidia.gpu) (gh)
  export/test_serdes.py::SerDesExportTestDynamismExpression::test_export_constraints_error_serdes
• pull / linux-focal-cuda12.1-py3.10-gcc9 / test (default, 5, 5, linux.4xlarge.nvidia.gpu) (gh)
  export/test_passes.py::TestPasses::test_runtime_assert_inline_constraints_for_item
• pull / linux-focal-cuda12.1-py3.10-gcc9-sm86 / test (default, 2, 5, linux.g5.4xlarge.nvidia.gpu) (gh)
  export/test_retraceability.py::RetraceExportTestDynamismExpression::test_export_constraints_error_retraceability
• pull / linux-focal-cuda12.1-py3.10-gcc9-sm86 / test (default, 3, 5, linux.g5.4xlarge.nvidia.gpu) (gh)
  export/test_serdes.py::SerDesExportTestDynamismExpression::test_export_constraints_error_serdes
• pull / linux-focal-cuda12.1-py3.10-gcc9-sm86 / test (default, 5, 5, linux.g5.4xlarge.nvidia.gpu) (gh)
  export/test_passes.py::TestPasses::test_runtime_assert_inline_constraints_for_item
• pull / linux-focal-py3.11-clang10 / test (crossref, 2, 2, linux.2xlarge) (gh)
  export/test_export.py::TestDynamismExpression::test_export_constraints_error
• pull / linux-focal-py3.11-clang10 / test (default, 1, 3, linux.2xlarge) (gh)
  export/test_serdes.py::SerDesExportTestDynamismExpression::test_export_constraints_error_serdes
• pull / linux-focal-py3.11-clang10 / test (default, 2, 3, linux.2xlarge) (gh)
  export/test_export.py::TestDynamismExpression::test_export_constraints_error
• pull / linux-focal-py3.11-clang10 / test (default, 3, 3, linux.2xlarge) (gh)
  export/test_passes.py::TestPasses::test_runtime_assert_inline_constraints_for_item
• pull / linux-focal-py3.12-clang10 / test (default, 1, 3, linux.2xlarge) (gh)
  export/test_retraceability.py::RetraceExportTestDynamismExpression::test_export_constraints_error_retraceability
• pull / linux-focal-py3.12-clang10 / test (default, 2, 3, linux.2xlarge) (gh)
  export/test_export.py::TestDynamismExpression::test_export_constraints_error
• pull / linux-focal-py3.12-clang10 / test (default, 3, 3, linux.2xlarge) (gh)
  export/test_passes.py::TestPasses::test_runtime_assert_inline_constraints_for_item
• pull / linux-focal-py3.8-clang10 / test (crossref, 2, 2, linux.2xlarge) (gh)
  export/test_export.py::TestDynamismExpression::test_export_constraints_error
• pull / linux-focal-py3.8-clang10 / test (default, 1, 3, linux.2xlarge) (gh)
  export/test_retraceability.py::RetraceExportTestDynamismExpression::test_export_constraints_error_retraceability
• pull / linux-focal-py3.8-clang10 / test (default, 2, 3, linux.2xlarge) (gh)
  export/test_export.py::TestDynamismExpression::test_export_constraints_error
• pull / linux-focal-py3.8-clang10 / test (default, 3, 3, linux.2xlarge) (gh)
  export/test_passes.py::TestPasses::test_runtime_assert_inline_constraints_for_item
• pull / linux-jammy-py3.10-clang15-asan / test (default, 3, 6, linux.4xlarge) (gh)
  export/test_serdes.py::SerDesExportTestDynamismExpression::test_export_constraints_error_serdes
• pull / linux-jammy-py3.10-clang15-asan / test (default, 5, 6, linux.4xlarge) (gh)
  export/test_passes.py::TestPasses::test_runtime_assert_inline_constraints_for_item
• pull / linux-jammy-py3.10-clang15-asan / test (default, 6, 6, linux.4xlarge) (gh)
  export/test_retraceability.py::RetraceExportTestDynamismExpression::test_export_constraints_error_retraceability
• pull / linux-jammy-py3.8-gcc11 / test (default, 2, 3, linux.2xlarge) (gh)
  export/test_export.py::TestDynamismExpression::test_export_constraints_error
• pull / linux-jammy-py3.8-gcc11 / test (default, 3, 3, linux.2xlarge) (gh)
  export/test_retraceability.py::RetraceExportTestDynamismExpression::test_export_constraints_error_retraceability

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

• pull / linux-focal-cuda12.1-py3.10-gcc9 / test (default, 2, 5, linux.4xlarge.nvidia.gpu) (gh) (similar failure)
  export/test_export.py::TestDynamismExpression::test_export_constraints_error
• pull / linux-jammy-py3.10-clang15-asan / test (default, 2, 6, linux.4xlarge) (gh) (similar failure)
  export/test_export.py::TestDynamismExpression::test_export_constraints_error

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

[ezyang]

welp, I guess export doesn't like it. cc @avikchaudhuri @pianpwk @tugsbayasgalan @angelayi