Fixes #322 (Change bf16 to amp_bf16)

mosaicml / examples

Fast and flexible reference benchmarks

Apache License 2.0

435 stars 124 forks source link

Open Taytay opened 8 months ago

Taytay commented 8 months ago

This fixes #322. When I was working through #440, I got bitten by this bug.