-
Notifications
You must be signed in to change notification settings - Fork 10
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
upgrade for bscvrq to support non-quant #113
base: paddlebox
Are you sure you want to change the base?
Conversation
// add for bscvrq | ||
// embedx_concate_filter:true && quant_ratio_valid=false && need_filter=true && embed_threshold_filter=false | ||
template <typename T> | ||
struct sum_pooling_concate<T, false, false, true> { |
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.
need_filter=true or false?
need_filter和embed_threshold_filter是否反了?
// embedx_concate_filter:true && quant_ratio_valid=false && need_filter=false | ||
template <typename T, bool use_cvm, bool clk_filter, bool need_filter, | ||
bool embed_threshold_filter, typename T2> | ||
struct do_sum_pooling_and_cvm<T, use_cvm, clk_filter, need_filter, false, embed_threshold_filter, true, T2> { |
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.
缺少了do_sum_pooling_and_cvm_with_large_dim的实现
// add for bscvrq | ||
// embedx_concate_filter:true && quant_ratio_valid=false && need_filter=false && embed_threshold_filter=false | ||
template <typename T> | ||
struct sum_pooling_concate<T, false, true, false, false> { |
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 for bscvrq | ||
// embedx_concate_filter:true && quant_ratio_valid=false && need_filter=true && embed_threshold_filter=false |
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.
这个跟上面的可以合并成一个,模版参数写成<false, need_filter, false>
// add for bscvrq | ||
// embedx_concate_filter:true && quant_ratio_valid=false && need_filter=true && embed_threshold_filter=true | ||
template <typename T> | ||
struct sum_pooling_concate<T, false, true, true> { |
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.
同理,这个跟现有的也可以合并,<quant_ratio_valid, true, true>
* fused_seqpool_cvm_with_conv support filter by threshold * add fill zero in fused_seqpool_cvm * add fused seq tensor && support transpose batch fc weight * fix fused query seq tensor compare case --------- Co-authored-by: mojingcj <[email protected]> Co-authored-by: jiaoxuewu <[email protected]> Co-authored-by: yuandong1998 <[email protected]> Co-authored-by: shangzhongbin <[email protected]>
PR types
PR changes
paddle/fluid/operators/fused/fused_seqpool_cvm_kernel.kps
Describe
支持非量化的分支,用于给bscvrq使用