support fp16 index_select (#50101)

0f173d5a · wangxiaoning · GitHub · ce4637c1 · 0f173d5a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/tensor/search.py python/paddle/tensor/search.py +1 -1

未找到文件。
--- 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(