Fix import error in torchtune due to bc breaking changes in torchao

[jerryzh168]

Summary: att

Differential Revision: D64261315

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 7 New Failures, 1 Cancelled Job, 2 Unrelated Failures

As of commit 2c559c6 with merge base 50d3ef1 ():

NEW FAILURES - The following jobs have failed:

• Build Docs / build_docs (3.11) (gh)
  sphinx.ext.autosummary.ImportExceptionGroup: no module named torchtune.training
• GPU tests / gpu_test (3.10, stable) (gh)
  E ImportError: cannot import name 'TensorCoreTiledLayout' from 'torchao.dtypes' (/home/ec2-user/actions-runner/_work/torchtune/torchtune/3/envs/test/lib/python3.10/site-packages/torchao/dtypes/__init__.py)
• GPU tests / gpu_test (3.9, stable) (gh)
  E ImportError: cannot import name 'TensorCoreTiledLayout' from 'torchao.dtypes' (/home/ec2-user/actions-runner/_work/torchtune/torchtune/3/envs/test/lib/python3.9/site-packages/torchao/dtypes/__init__.py)
• Recipe Tests / recipe_test (3.11) (gh)
  E ImportError: cannot import name 'TensorCoreTiledLayout' from 'torchao.dtypes' (/usr/share/miniconda3/envs/test/lib/python3.11/site-packages/torchao/dtypes/__init__.py)
• Recipe Tests / recipe_test (3.9) (gh)
  E ImportError: cannot import name 'TensorCoreTiledLayout' from 'torchao.dtypes' (/usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torchao/dtypes/__init__.py)
• Unit Test / unit_tests (3.10) (gh)
  E ImportError: cannot import name 'TensorCoreTiledLayout' from 'torchao.dtypes' (/usr/share/miniconda3/envs/test/lib/python3.10/site-packages/torchao/dtypes/__init__.py)
• Unit Test / unit_tests (3.9) (gh)
  E ImportError: cannot import name 'TensorCoreTiledLayout' from 'torchao.dtypes' (/usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torchao/dtypes/__init__.py)

CANCELLED JOB - The following job was cancelled. Please retry:

• Recipe Tests / recipe_test (3.10) (gh)

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

• Unit Test / unit_tests (3.11) (gh) (similar failure)
  ##[error]The operation was canceled.

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

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

• GPU tests / gpu_test (3.11, stable) (gh) (trunk failure)
  ##[error]The operation was canceled.

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

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D64261315

[jerryzh168]

real fix in #1812