-
Notifications
You must be signed in to change notification settings - Fork 9
11.22 #101
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
base: wlt
Are you sure you want to change the base?
Conversation
test_op("upsampling"); // formal & cpu pass | ||
test_op("dense"); // formal & cpu pass | ||
int main(int argc, char* argv[]) { | ||
ctx = APIDevTypeMap.at(stoi(argv[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.
Add more information about arguments, such as the meaning, format, etc.
tests/test_op.cc
Outdated
//test_op("non_max_suppression"); // formal & cpu pass | ||
// test_op("broadcast_sub"); // formal & cpu pass | ||
// test_op("broadcast_add"); // formal & cpu pass | ||
// test_op("broadcast_mul"); // formal & cpu pass | ||
// test_op("broadcast_max"); // formal & cpu pass | ||
// test_op("broadcast_div"); // formal & cpu pass | ||
// test_op("broadcast_greater"); // formal & cpu pass |
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.
Update the GPU pass information like your test documentation.
tests/test_op2.cc
Outdated
return result(tshape, tdata, params, attr)[0]; | ||
} | ||
|
||
NDArray dense(NDArray const& data, NDArray const& weight, string attr_str, NDArray* bias = nullptr) |
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.
Support three typed-APIs:
- NDArray
- DLTensor
- int32_t *
tests/test_op2.cc
Outdated
return res; | ||
} | ||
|
||
NDArray conv2d(NDArray const& data, NDArray const& weight, string attr_str, NDArray* bias = nullptr) |
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.
use typed-attributes in parameters, instead of attr_str.
No description provided.