From 29d24dbb800f00d3b4feac99a8960f6387856528 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 26 Aug 2021 19:54:59 +0800
Subject: [PATCH] fix(mge/function): fix interpolate unsupport fp16 error

GitOrigin-RevId: 7fc6271986ce94b496e7be42b1d4e8500a3ca921
---
 imperative/python/megengine/functional/vision.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/imperative/python/megengine/functional/vision.py b/imperative/python/megengine/functional/vision.py
index f067f08f..b713a829 100644
--- a/imperative/python/megengine/functional/vision.py
+++ b/imperative/python/megengine/functional/vision.py
@@ -582,7 +582,8 @@ def interpolate(
             "nearest": "nearest",
             "bicubic": "cubic",
         }
-
+        if inp.dtype == np.float16:
+            inp = inp.astype("float32")
         op = builtin.Resize(imode=mode_map[mode], format="NCHW")
         shape = astensor1d(dsize, inp, dtype="int32", device=inp.device)
         (ret,) = apply(op, inp, shape)