Merge pull request #241 from mkolod/fp32_interp

Run interpolation in fp32 because it's faster

Merge pull request #241 from mkolod/fp32_interp
Run interpolation in fp32 because it's faster
dc8f400f · ngimel · GitHub · 3f87614f · 47e9cae5 · dc8f400f
Unverified Commit dc8f400f authored Apr 04, 2019 by ngimel Committed by GitHub Apr 04, 2019
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',