This repository has been archived by the owner on Jan 24, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 115
op unittest for cbrt/ceil/cholesky/concat/constant/fill_constant #1495
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Thanks for your contribution! |
描述From #1378 cbrt 算子。 算子类型
Test Cases Checklist张量维度
special shape挑选 2D/3D/4D 张量中的一个,测试下面的特殊情况。
张量数据类型
广播
算子属性无。 |
描述From #1378 ceil 算子。 算子类型
Test Cases Checklist张量维度
special shape挑选 2D/3D/4D 张量中的一个,测试下面的特殊情况。
张量数据类型
广播
算子属性无。 |
描述From #1378 cholesky 算子。 算子类型
Test Cases Checklist张量维度cholesky 支持
special shape挑选 2D/3D/4D 张量中的一个,测试下面的特殊情况。
张量数据类型
广播
算子属性
|
描述From #1378 concat 算子。 算子类型
Test Cases Checklist张量维度
special shape挑选 2D/3D/4D 张量中的一个,测试下面的特殊情况。
张量数据类型
广播
算子属性
|
描述From #1378 constant 算子。 算子类型
Test Cases Checklist张量维度
special shape挑选 2D/3D/4D 张量中的一个,测试下面的特殊情况。
张量数据类型
广播
算子属性
|
zzk0
commented
Jun 2, 2023
zzk0
commented
Jun 2, 2023
zzk0
changed the title
op unittest for cbrt/ceil/cholesky/concat/constant
op unittest for cbrt/ceil/cholesky/concat/constant/fill_constant
Jun 3, 2023
描述From #1378 fill_constant 算子。 算子类型
Test Cases Checklist张量维度
special shape挑选 2D/3D/4D 张量中的一个,测试下面的特殊情况。
张量数据类型
广播
算子属性
#define PY_REGISTER_FILLCONSTANT_OP(TYPE__) \
.def("fill_constant", \
static_cast<Variable (NetBuilder::*)( \
const std::vector<int> &, TYPE__, const std::string &, bool)>( \
&NetBuilder::template FillConstant<TYPE__>), \
py::arg("shape"), \
py::arg("value"), \
py::arg("name") = "", \
py::arg("force_cpu") = false)
EXPAND_CINN_SUPPORT_TYPE(PY_REGISTER_FILLCONSTANT_OP)
#undef PY_REGISTER_FILLCONSTANT_OP
.def("fill_constant",
static_cast<Variable (NetBuilder::*)(
const std::vector<int> &, float, const std::string &, const std::string &, bool)>(
&NetBuilder::FillConstant),
py::arg("shape"),
py::arg("value"),
py::arg("name") = "",
py::arg("dtype"),
py::arg("force_cpu") = false)
.def("fill_constant",
static_cast<Variable (NetBuilder::*)(
const std::vector<int> &, const std::string &, const std::string &, const std::string &, bool)>(
&NetBuilder::FillConstant),
py::arg("shape"),
py::arg("value"),
py::arg("name") = "",
py::arg("dtype"),
py::arg("force_cpu") = false) |
thisjiang
reviewed
Jun 5, 2023
thisjiang
approved these changes
Jun 8, 2023
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.
LGTM
Closed
Open
BiynXu
added a commit
that referenced
this pull request
Jun 19, 2023
* op unittest for cbrt/ceil/cholesky/concat/constant/fill_constant (#1495) * op unittest for cbrt * op unittest for ceil * op unittest for cholesky * op unittest for concat * op unittest for constant * add 4d test for constant op * fix ci * op unittest for fill_constant * op unittest for fill_constant * refine * fix(schedule): fix SimpleComputeAt primitive (#1504) * Fix reduce cast schedule bug (#1512) * fix(fuse): fix reduce cast schedule bug * test(fuse): add unittest for reduce_cast subgroup * Refactor some op tests and fix bugs (#1515) * Add depthwise_conv2d op test * Refactor log op test * Refactor round op test and fix bugs * Only test depthwise_conv2d in cuda_cudnn * op unittest for repeat/arange/reverse/elementwise_add_grad/flip (#1514) * op unittest for repeat op * add repeat frontend * op unittest for repeat * op unittest for arange * op unittest for reverse * format & remove old add op test * op unittest for flipe && remove redundant flip implementation * remove test_add_op_new.py * update reverse * Refactor some op tests and fix bugs (#1513) * Refactor op isclose test * Refactor op logical_right_shift and add more dtypes support * Refactor pow op test and fix bugs * Refactor lookup_table op test * Add logical_right_shift host function proto * Improve isclose test case * Fixed jitify commit to prevent header file conflicts (#1522) * Fixed jitify commit to prevent header file conflicts * Set random seed for debug floor_divide * Avoid oom error * Just for debug ci * Fix floor_divide error when input dtype is int * Fix bugs and add more tests for floor_divide * Experimental PR for the first OP to clean old schedule (#1524) --------- Co-authored-by: zzk0 <[email protected]> Co-authored-by: Fisher <[email protected]> Co-authored-by: Huihuang Zheng <[email protected]>
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
描述
From #1378
给 cbrt/ceil/cholesky/concat/constant 算子添加单元测试。