Skip to content

Commit 77a8785

Browse files
modify windows bug, input dim was change in win
1 parent 19da3dd commit 77a8785

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddle/phi/kernels/gpu/set_value_kernel.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ void SetTensorValueKernelV2(const Context& dev_ctx,
4242
const std::vector<int64_t>& none_axes,
4343
DenseTensor* out) {
4444
auto in_dims = in.dims();
45+
auto meta = in.meta();
4546
std::vector<int64_t> starts_local = starts.GetData();
4647
std::vector<int64_t> ends_local = ends.GetData();
4748
std::vector<int64_t> steps_local = steps.GetData();
@@ -107,7 +108,7 @@ void SetTensorValueKernelV2(const Context& dev_ctx,
107108
new_out_stride,
108109
output_offset,
109110
out);
110-
out->set_meta(in.meta());
111+
out->set_meta(meta);
111112
}
112113

113114
template <typename T, typename Context>

0 commit comments

Comments
 (0)