add ability to calculate amax in tiles

[drisspg]

trafficstars

Stacked PRs:

• #709
• #707
• ->#706

---

add ability to calculate amax in tiles

ghstack-source-id: 83ccec3ec66f30b9d75146d0fc7b1137ea7574c4 Pull Request resolved: https://github.com/pytorch/ao/pull/682

[pytorch-bot[bot]]

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/706

• :page_facing_up: Preview Python docs built from this PR

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

:x: 1 New Failure

As of commit 655fc4c6c6762f21dc25f7ae26a178e202597455 with merge base ac8ce4ceb548c264a2f508f7b6a1f413d8d4454c ():

NEW FAILURE - The following job has failed:

• Run Float8 Tests / test (SM-89, amz2023.linux.g6.4xlarge.experimental.nvidia.gpu, --pre torch --index-url https://do... / linux-job (gh) test/float8/test_inference_flows.py::TestFP8TrainToFP8LinearInference::test_fp8_save_and_load[dtype1]

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