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

[Feature](mluOpSgetrf2): add LU parameter #105

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

Chuancysun
Copy link
Collaborator

add LU parameter

@shunshen93
Copy link
Collaborator

Feature: add cholesky parameter #95
仿照这个修改下标题吧

@@ -1119,6 +1121,11 @@ message FFTParam {
optional float scale_factor = 4 [default = 1.0];
}

// param to call Sgetrf2()
message Sgetrf2Param {
Copy link
Collaborator

Choose a reason for hiding this comment

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

sgetrf2改成lu吧

Copy link
Collaborator

Choose a reason for hiding this comment

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

sgetrf2改成xgetrf,不是LU

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

已修改

@Chuancysun Chuancysun changed the title add LU parameter [Feature](mluOpSgetrf2): add LU parameter Nov 20, 2024
@@ -1119,6 +1121,11 @@ message FFTParam {
optional float scale_factor = 4 [default = 1.0];
}

// param to call Sgetrf2()
message Sgetrf2Param {
optional int32 mode =1 [default = 1];
Copy link
Collaborator

Choose a reason for hiding this comment

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

缺少inplace的参数

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

这里不支持inplace

Copy link
Collaborator

Choose a reason for hiding this comment

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

https://github.com/Reference-LAPACK/lapack/blob/master/SRC/dgetrf2.f#L69
getrf这个算子,应该是支持原位修改输入的

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

这个我们在最初时是做的原位的,但后来对接时需求更改为了非原位的

@@ -1119,6 +1121,11 @@ message FFTParam {
optional float scale_factor = 4 [default = 1.0];
}

// param to call Sgetrf2()
message Sgetrf2Param {
optional int32 mode =1 [default = 1];
Copy link
Collaborator

Choose a reason for hiding this comment

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

https://github.com/Reference-LAPACK/lapack/blob/master/SRC/dgetrf2.f#L69
getrf这个算子,应该是支持原位修改输入的

@@ -256,7 +256,8 @@ enum OpType {
DYNAMIC_POINT_TO_VOXEL_BACKWARD = 404;
DYNAMIC_POINT_TO_VOXEL_FORWARD = 405;
EXPAND = 5;
FFT = 65536;
FFT = 65536;
SGETRF2 = 65537;
Copy link
Collaborator

@shunshen93 shunshen93 Dec 3, 2024

Choose a reason for hiding this comment

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

XGETRF

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants