Skip to content

expose stride from disc ral to torch tensor#1124

Open
Yancey0623 wants to merge 3 commits intoalibaba:mainfrom
Yancey0623:expose_layout_to_torch_ral
Open

expose stride from disc ral to torch tensor#1124
Yancey0623 wants to merge 3 commits intoalibaba:mainfrom
Yancey0623:expose_layout_to_torch_ral

Conversation

@Yancey0623
Copy link
Collaborator

No description provided.

Copy link

@qiuxiafei qiuxiafei left a comment

Choose a reason for hiding this comment

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

LGTM~

torch::from_blob(
const_cast<void*>(out_buf->data()), out_buf->shape(), option)
.clone();
out_tensor = torch::from_blob(
Copy link
Collaborator

Choose a reason for hiding this comment

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

Don't you need to copy the underline buffer? @wyzero Pls double check if it is safe.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants

Comments