#618 Add narrow_copy, cholesky_solve

Merged
zoulq merged 1 commits from dev0718 into master 10 months ago
暂未完成GPU和Ascend上的验证
liuzhuoran2333 changed title from Add narrow_copy, cholesky_solve to [WIP]Add narrow_copy, cholesky_solve 10 months ago
zoulq reviewed 10 months ago
@@ -1056,0 +1069,4 @@
a_ms = ms_torch.mm(a_ms, a_ms.t()) + 1e-05 * ms_torch.eye(3)
u_ms = ms_torch.cholesky(a_ms)
ms_out1 = ms_torch.cholesky_solve(u_ms, b_ms)
ms_out2 = ms_torch.cholesky_solve(u_ms, b_ms, True)
zoulq commented 10 months ago
拆成两个用例,数据类型和入参交叉叠加
liuzhuoran2333 commented 10 months ago
fixed
zoulq reviewed 10 months ago
SupportedList_en.md
@@ -537,6 +539,7 @@ English | [简体中文](SupportedList.md)
| Tensor.nanmean | Supported | |
| Tensor.nansum | Supported | |
| Tensor.narrow | Supported | |
| Tensor.narrow_copy | Supported | |
zoulq commented 10 months ago
还有个torch.narrow_copy接口一起开发
liuzhuoran2333 commented 10 months ago
fixed
liuzhuoran2333 changed title from [WIP]Add narrow_copy, cholesky_solve to Add narrow_copy, cholesky_solve 10 months ago
zoulq reviewed 10 months ago
@@ -1056,0 +1088,4 @@
ms_out1 = ms_torch.cholesky_solve(a_ms, b_ms)
ms_out2 = ms_torch.cholesky_solve(a_ms, b_ms, True)
param_compare(torch_out1, ms_out1)
param_compare(torch_out2, ms_out2)
zoulq commented 10 months ago
按照之前讨论的策略,fP64用例可以再精简一些
liuzhuoran2333 commented 10 months ago
用例已精简
Erpim reviewed 10 months ago
testing/ut/pytorch/functional/test_math.py
@@ -2673,0 +2678,4 @@
out1_ms = x1_ms.narrow_copy(0, 1, 2)
out2_pt = x1_pt.narrow_copy(-1, 2, 3)
out2_ms = x1_ms.narrow_copy(-1, 2, 3)
param_compare(out1_pt, out1_ms)
Erpim commented 10 months ago
ops.narrow调用的Slice算子,添加一个测试用例,对output修改后,看下原始tensor会不会一起被修改。
liuzhuoran2333 commented 10 months ago
已新增用例看护
liuzhuoran2333 commented 10 months ago
Poster
narrow_copy用例已重新整改
zoulq merged commit dba593129b into master 10 months ago
liuzhuoran2333 deleted branch dev0718 10 months ago
The pull request has been merged as dba593129b.
Sign in to join this conversation.
No reviewers
No Label
No Milestone
No Assignees
3 Participants
Notifications
Due Date

No due date set.

Dependencies

This pull request currently doesn't have any dependencies.

Loading…
There is no content yet.