Skip to content

Commit e07e5a4

Browse files
【set_value】temp fix Set tensor cuda error (#73251)
* add float8 support for concat * add concat grad support fp8 * set_value support fp8 * modify python api * modify ci test * fix set_tensor dst place is undefined bug * clear * find real bug * find real bug
1 parent 7a5f15c commit e07e5a4

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

python/paddle/base/dygraph/tensor_patch_methods.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -258,12 +258,20 @@ def set_value(
258258
self.value().process_mesh,
259259
self.value().placements,
260260
)
261-
if isinstance(value, paddle.Tensor):
261+
if (
262+
isinstance(value, paddle.Tensor)
263+
and value.is_contiguous()
264+
and self.value().is_contiguous()
265+
):
262266
self.value().set_tensor(value)
263267
else:
264268
self.value().get_tensor().set(value.get_tensor())
265269
return
266-
if isinstance(value, paddle.Tensor):
270+
if (
271+
isinstance(value, paddle.Tensor)
272+
and value.is_contiguous()
273+
and self.value().is_contiguous()
274+
):
267275
self.value().set_tensor(value)
268276
else:
269277
self.value().get_tensor().set(

0 commit comments

Comments
 (0)