Skip to content

GitLab

Unverified Commit dc8f400f authored by ngimel's avatar ngimel Committed by GitHub
Browse files

Merge pull request #241 from mkolod/fp32_interp 

Run interpolation in fp32 because it's faster
parents 3f87614f 47e9cae5
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
apex/amp/lists/functional_overrides.py
View file @ dc8f400f
...@@ -27,6 +27,10 @@ FP16_FUNCS = [ ...@@ -27,6 +27,10 @@ FP16_FUNCS = [
] ]
FP32_FUNCS = [ FP32_FUNCS = [
# Interpolation/Upsampling
'interpolate',
# Pointwise # Pointwise
'softplus', 'softplus',
'softmin', 'softmin',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment