Run interpolation in fp32 because it's faster

47e9cae5 · Marek Kolodziej · 3f87614f · 47e9cae5
Commit 47e9cae5 authored Apr 04, 2019 by Marek Kolodziej
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

apex/amp/lists/functional_overrides.py apex/amp/lists/functional_overrides.py +4 -0

No files found.
--- a/apex/amp/lists/functional_overrides.py
+++ b/apex/amp/lists/functional_overrides.py
@@ -27,6 +27,10 @@ FP16_FUNCS = [
 ]

 FP32_FUNCS = [
+
+    # Interpolation/Upsampling
+    'interpolate',
+
    # Pointwise
    'softplus',
    'softmin',