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 9b1dce1df86..3ec79b55b0b 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