fix (#51552)

c3f8ba9b · wangxiaoning · GitHub · 2876f6f8 · c3f8ba9b · c3f8ba9b
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/phi/kernels/cpu/index_select_grad_kernel.cc paddle/phi/kernels/cpu/index_select_grad_kernel.cc +1 -0

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

未找到文件。
--- a/paddle/phi/kernels/cpu/index_select_grad_kernel.cc
+++ b/paddle/phi/kernels/cpu/index_select_grad_kernel.cc
@@ -59,5 +59,6 @@ PD_REGISTER_KERNEL(index_select_grad,
                   phi::IndexSelectGradKernel,
                   float,
                   double,
+                   phi::dtype::bfloat16,
                   int,
                   int64_t) {}
--- a/python/paddle/tensor/search.py
+++ b/python/paddle/tensor/search.py
@@ -350,7 +350,7 @@ def index_select(x, index, axis=0, name=None):
        check_variable_and_dtype(
            x,
            'x',
-            ['bfloat16', 'float16', 'float32', 'float64', 'int32', 'int64'],
+            ['uint16', 'float16', 'float32', 'float64', 'int32', 'int64'],
            'paddle.tensor.search.index_select',
        )
        check_variable_and_dtype(