ao
ao copied to clipboard
add ability to calculate amax in tiles
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
: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.