-
Notifications
You must be signed in to change notification settings - Fork 5.7k
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
"add data layout" #6955
"add data layout" #6955
Conversation
@@ -115,6 +117,10 @@ class Tensor { | |||
|
|||
inline void check_memory_size() const; | |||
|
|||
inline DataLayout layout() const { return layout_; } | |||
|
|||
inline void set_layout(const DataLayout layout) { layout_ = layout; } |
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.
Bind this to pybind?
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.
Done.
Seem CI failed on CopyTensor ops, segmentfault |
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++
fix #6765