Skip to content
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

Infer consistent tensor meta #5118

Merged
merged 178 commits into from
Jul 1, 2021
Merged

Conversation

lixinqi
Copy link
Contributor

@lixinqi lixinqi commented Jun 7, 2021

使用ConsistentTensorInferCache来加速推导eager conssitent tensor的元信息。

lixinqi and others added 30 commits May 12, 2021 11:10
@clackhan clackhan requested review from oneflow-ci-bot and removed request for oneflow-ci-bot June 28, 2021 02:11
@oneflow-ci-bot oneflow-ci-bot removed their request for review June 28, 2021 02:56
template<>
struct hash<oneflow::one::ConsistentTensorMeta> final {
size_t operator()(const oneflow::one::ConsistentTensorMeta& other) const {
return other.CalcHashValue();
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

CalcHashValuehash_value 两种命名有什么区别

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

CalcHashValue具有计算过程,hash_value是从对象上拿hash已经存在的值

@clackhan clackhan requested a review from oneflow-ci-bot June 29, 2021 10:33
clackhan added 2 commits July 1, 2021 11:07
…infer_consistent_tensor_meta

Conflicts:
	oneflow/core/framework/op_expr.h
	oneflow/core/framework/tensor.h
	oneflow/core/framework/tensor_impl.h
…infer_consistent_tensor_meta

Conflicts:
	oneflow/core/framework/tensor_impl.cpp
@clackhan clackhan requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 1, 2021 03:11
@oneflow-ci-bot oneflow-ci-bot removed their request for review July 1, 2021 03:11
@oneflow-ci-bot oneflow-ci-bot self-requested a review July 1, 2021 03:12
@clackhan clackhan requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 1, 2021 03:52
@oneflow-ci-bot oneflow-ci-bot requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 1, 2021 04:43
const UserOpExpr& user_op_expr, const std::vector<BlobDesc>& blob_descs,
std::vector<ParallelDistributionInferHint>* hints) const;

static Maybe<ConsistentTensorMetaInferArgs> New() {
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

把 New 和 Init 合在一起吧,static Maybe<ConsistentTensorMetaInferArgs> New(const TensorTuple& input_tensors, Symbol<PlacementScope> placement_scope, const AttrMap& attrs);

@oneflow-ci-bot oneflow-ci-bot removed their request for review July 1, 2021 06:28
@oneflow-ci-bot oneflow-ci-bot merged commit c3238bf into master Jul 1, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants