PR #8498 fix batch_norm amp autocast

fix batch_norm amp autocast #8498

yaochengji merged 2 commits into pytorch:master from yaochengji:chengji/fix-amp

fix batch_norm amp autocast

0b0b91b1

enable bfloat16 amp test only for TPU and GPU

8c2bfa1f

qihqi requested a review from

qihqi 1 year ago

qihqi approved these changes on 2025-01-06

yaochengji merged 1cdb1eff into master 1 year ago

Reviewers

qihqi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone