-
Notifications
You must be signed in to change notification settings - Fork 27
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
base: master
Are you sure you want to change the base?
Conversation
mlu_op_test.proto
Outdated
@@ -1119,6 +1121,11 @@ message FFTParam { | |||
optional float scale_factor = 4 [default = 1.0]; | |||
} | |||
|
|||
// param to call Sgetrf2() | |||
message Sgetrf2Param { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
sgetrf2改成lu吧
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
sgetrf2改成xgetrf,不是LU
There was a problem hiding this comment.
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]; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
缺少inplace的参数
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里不支持inplace
There was a problem hiding this comment.
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这个算子,应该是支持原位修改输入的
There was a problem hiding this comment.
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]; |
There was a problem hiding this comment.
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; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
XGETRF
add LU parameter