已合并
[sync] PR-37817: Fix NPU tensor rebuild requires_grad load #37987
ascend-robot创建于 6月9日
[sync] PR-37817: Fix NPU tensor rebuild requires_grad load #37987
已合并
共 1 个文件变更+0-1
| @@ -20,7 +20,6 @@ def _rebuild_npu_tensor(storage, storage_offset, size, stride, requires_grad, ba | |||
| 20 | (0,), | 20 | (0,), |
| 21 | dtype=storage.dtype, | 21 | dtype=storage.dtype, |
| 22 | device=storage._untyped_storage.device, | 22 | device=storage._untyped_storage.device, |
| 23 | - requires_grad=requires_grad, | ||
| 24 | ) | 23 | ) |
| 25 | tensor.set_(storage, storage_offset, size, stride) | 24 | tensor.set_(storage, storage_offset, size, stride) |
| 26 | tensor.requires_grad = requires_grad | 25 | tensor.requires_grad = requires_grad |