Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[WIP]add mluop cholesky #1146

Open
wants to merge 45 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
45 commits
Select commit Hold shift + click to select a range
fc47e70
complete the float type cholesky operator
dglr Apr 16, 2024
77db74c
[WIP]add mluop cholesky
dglr Apr 30, 2024
0e9a1f8
add cholesky doc
dglr May 23, 2024
6051cce
modify mathematical formula
dglr May 27, 2024
86a2c41
add complex type
dglr Jun 7, 2024
efa3d08
finish complex batch
dglr Jun 28, 2024
4872a42
fix ang bugs
dglr Jul 19, 2024
0f12676
fix nram workspace, update doc
dglr Jul 23, 2024
86ceaba
add pseudocode
dglr Jul 25, 2024
beb7e53
add comments
dglr Jul 25, 2024
40f62ba
add index.rst
dglr Jul 25, 2024
d935fb9
format code
dglr Jul 25, 2024
e42d270
[Fix](mluOpCholesky): fix format
dglr Aug 12, 2024
bfcf2b2
[Fix](mluOpCholesky): add mluoplog when sqrt
dglr Aug 15, 2024
e4f330b
[Fix](mluOpCholesky): reset workspace
dglr Aug 15, 2024
9eb9dc4
[Fix](mluOpCholesky): rename getworkspace size function
dglr Aug 15, 2024
76631ee
[Fix](mluOpCholesky): rewrite description in mlu_op
dglr Aug 15, 2024
23acef7
[Docs](mluOpCholesky): update docs
dglr Aug 15, 2024
40661cc
[Fix](mluOpCholesky): del printf
dglr Aug 15, 2024
4d34d54
[Docs](mluOpCholesky): rewrite Conjugate transpose symbol
dglr Aug 15, 2024
fc1a0ac
[Fix](mluOpCholesky): format
dglr Aug 24, 2024
b0d5b6e
[Fix](mluOpCholesky): add layout check
dglr Sep 16, 2024
9f7dcd5
[Fix](mluOpCholesky): fix mem check
dglr Sep 16, 2024
7888578
[Docs](mluOpCholesky): add test doc
dglr Sep 16, 2024
435f829
[Docs](mluOpCholesky): add coverage test
dglr Sep 20, 2024
4f9a1af
[Fix](mluOpCholesky): add dimension equals test
dglr Sep 20, 2024
43bbe67
[Fix](mluOpCholesky): add coverage function
dglr Sep 21, 2024
976a88b
[Fix](mluOpCholesky): test
Nov 13, 2024
284e6e8
[Fix](mluOpCholesky): test
Nov 14, 2024
65705ea
[Fix](mluOpCholesky): resolve conflict
Nov 14, 2024
4d18629
[Fix](mluOpCholesky): fix magic number
Nov 14, 2024
5ae8c94
[Fix](mluOpCholesky): cut one branch
Nov 14, 2024
2b5822d
[Fix](mluOpCholesky): add policy func
Nov 30, 2024
ecc80ec
[Fix](mluOpCholesky): rename variables
Dec 1, 2024
0938f55
[Fix](mluOpCholesky): add some comments
Dec 1, 2024
3213104
[Fix](mluOpCholesky): mv cnnl to cpp
Dec 1, 2024
a9c9db6
[Fix](mluOpCholesky): add new memcpy
Dec 4, 2024
876ebc2
[Fix](mluOpCholesky): remove useless sync
Dec 4, 2024
94664e5
[Fix](mluOpCholesky): remove cnrtmemcpy cnrtqueuesync
Dec 8, 2024
2c96f6f
[Fix](mluOpCholesky): update cholesky_test
Dec 8, 2024
9718b02
[Fix](mluOpCholesky): fix bugs
Dec 9, 2024
a472285
[Docs](mluOpCholesky): update doc
Dec 9, 2024
5d35eca
[Fix](mluOpCholesky): fix sync bugs
Dec 9, 2024
55ff179
[Docs](mluOpCholesky): update doc
Dec 9, 2024
b52a27a
[Fix](mluOpCholesky): add param check
Dec 10, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file added docs/design_docs/cholesky/32_128性能分析.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
537 changes: 537 additions & 0 deletions docs/design_docs/cholesky/cholesky.md

Large diffs are not rendered by default.

Binary file added docs/design_docs/cholesky/coverage.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/coverage_error.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/divide.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/gemm.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/potrf.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/recur_p1.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/recur_p2.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/syrk.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/timeline.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/design_docs/cholesky/trsm.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
7 changes: 7 additions & 0 deletions docs/user_guide/9_operators/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -793,3 +793,10 @@ mluOpLgamma

- ``x`` 为输入张量。

.. cholesky::
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这里的.. cholesky::是多写了吧?

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
.. cholesky::
.. _cholesky:


mluOpCholesky
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

有公式的话,需要补充下公式。

---------------
执行 Cholesky 分解,将一个正定矩阵分解为其下三角矩阵(L)或其转置的上三角矩阵(U),具体分解为上三角或下三角取决于参数``upper``。
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
执行 Cholesky 分解,将一个正定矩阵分解为其下三角矩阵(L)或其转置的上三角矩阵(U),具体分解为上三角或下三角取决于参数``upper``。
执行 Cholesky 分解,将一个正定矩阵分解为下三角矩阵(L)或转置的上三角矩阵(U)。分解为上三角还是下三角取决于参数 ``upper``。

可以把里面的“其”删除吗?


该算子包含7个输入:handle 为操作句柄,input_desc 与 d_input 分别描述并提供输入矩阵的信息;两个输出:output_desc 与 d_output 分别描述并存储输出矩阵的信息;此外,还包含一个布尔参数 upper,用于指定输出是上三角还是下三角矩阵,以及一个 workspace 用于临时存储计算过程中的数据。
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
该算子包含7个输入:handle 为操作句柄,input_desc 与 d_input 分别描述并提供输入矩阵的信息;两个输出:output_desc 与 d_output 分别描述并存储输出矩阵的信息;此外,还包含一个布尔参数 upper用于指定输出是上三角还是下三角矩阵,以及一个 workspace 用于临时存储计算过程中的数据。
该算子包含7个输入:handle 为操作句柄,input_desc 与 d_input 分别描述并提供输入矩阵的信息;两个输出:output_desc 与 d_output 分别描述并存储输出矩阵的信息;此外,还包括参数 ``upper`` 用于指定输出是上三角还是下三角矩阵,以及 ``workspace`` 用于临时存储计算过程中的数据。

7个输入,这里只说明了3个(handle,input_desc 与 d_input)?upper,workspace也是输入的话,建议放在一起说明。

Loading
Loading