use float buffers to accumulate for BFloat16 upsample backward

pytorch · Dec 23, 2022 · 3905d48 · 3905d48
1 parent be41a3f
commit 3905d48
Show file tree

Hide file tree

Showing 3 changed files with 680 additions and 50 deletions.
diff --git a/aten/src/ATen/native/UpSample.h b/aten/src/ATen/native/UpSample.h
@@ -251,7 +251,7 @@ template <typename scalar_t>
 static inline scalar_t compute_scales_value(
     const c10::optional<double> scale,
     int64_t input_size,
-    int64_t output_size) { 
+    int64_t output_size) {
       // see Note [compute_scales_value]
       // FIXME: remove magic > 0 after we ensure no models were serialized with -1 defaults.
       return (scale.has_value() && scale.value() > 0.)