提交 a0bc13b4 编写于 作者: L lixinqi

Merge branch 'dev_python' of https://github.com/Oneflow-Inc/oneflow into dev_python

......@@ -10,23 +10,12 @@ class BroadcastFloorModOp final : public BroadcastBinaryOp {
private:
const PbMessage& GetCustomizedConf() const override;
Maybe<void> VirtualGetSbpSignatures(
const std::function<Maybe<const BlobDesc*>(const std::string&)>& LogicalBlobDesc4Ibn,
SbpSignatureList* sbp_sig_list) const override;
};
const PbMessage& BroadcastFloorModOp::GetCustomizedConf() const {
return op_conf().broadcast_floor_mod_conf();
}
Maybe<void> BroadcastFloorModOp::VirtualGetSbpSignatures(
const std::function<Maybe<const BlobDesc*>(const std::string&)>& LogicalBlobDesc4Ibn,
SbpSignatureList* sbp_sig_list) const {
SbpSignatureBuilder().PartialSum("a").Broadcast("b").PartialSum("out").Build(
sbp_sig_list->mutable_sbp_signature()->Add());
return Maybe<void>::Ok();
}
REGISTER_OP(OperatorConf::kBroadcastFloorModConf, BroadcastFloorModOp);
} // namespace oneflow
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册