From 0f173d5a62b41355b457c6434f6c1c1bf1ae4a79 Mon Sep 17 00:00:00 2001 From: wangxiaoning <71813629+wangxn12138@users.noreply.github.com> Date: Tue, 31 Jan 2023 14:11:37 +0800 Subject: [PATCH] support fp16 index_select (#50101) --- python/paddle/tensor/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/tensor/search.py b/python/paddle/tensor/search.py index 9b1dce1df8..3ec79b55b0 100755 --- a/python/paddle/tensor/search.py +++ b/python/paddle/tensor/search.py @@ -326,7 +326,7 @@ def index_select(x, index, axis=0, name=None): check_variable_and_dtype( x, 'x', - ['float32', 'float64', 'int32', 'int64'], + ['float16', 'float32', 'float64', 'int32', 'int64'], 'paddle.tensor.search.index_select', ) check_variable_and_dtype( -- GitLab