-
Notifications
You must be signed in to change notification settings - Fork 157
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
[onert] Add lifetime enum to LayerScopeTensor #13861
Conversation
enum class LayerScopeTensorLifeTime : unsigned char | ||
{ | ||
BACKWARD, // alive during backward() | ||
FORWARD_TO_BACKWARD, // alive from forward() to backward() | ||
}; |
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.
Lifetime info will be used to plan and allocate a buffer for LayerScopeTensor.
This PR adds lifetime info to the LayerScopeTensor. Lifetime info will be used to plan and allocate buffer for LayerScopeTensor. ONE-DCO-1.0-Signed-off-by: seunghui youn <[email protected]>
c1cc989
to
fa33a00
Compare
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
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
@@ -26,6 +26,12 @@ namespace backend | |||
namespace train | |||
{ | |||
|
|||
enum class LayerScopeTensorLifeTime : unsigned char |
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.
(out of my curiosity)
Why do you set a type for this enumaration class? What is the advantages?
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.
AFAIK, if we didn't set a type(enum-base
) for enum class, the enum element takes 4 bytes by default.
In this case, I'd like to reduce the size of enum element. (default : 4bytes -> unsigned char 1bytes)
This PR adds lifetime info to the LayerScopeTensor. Lifetime info will be used to plan and allocate buffer for LayerScopeTensor.
ONE-DCO-1.0-Signed-off-by: seunghui youn [email protected]
related : #13282
draft : #13486