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

Adds Tensors objects to DLA #54

Merged
merged 5 commits into from
Aug 1, 2024
Merged

Conversation

vilukissa68
Copy link
Collaborator

#51 Turned out to be way too big to review, so I split tensor functionality to this PR to reduce the amount of code to review.

Tensors are used to abstract the shape of data. This is useful since DLA expected a different data shape than what TVM does. Also since different models might use other data formats it is sensible to generalize tensors to all possible data shapes.

@vilukissa68 vilukissa68 mentioned this pull request Jul 31, 2024
@vilukissa68 vilukissa68 requested a review from hegza July 31, 2024 07:05
Copy link
Contributor

@hegza hegza left a comment

Choose a reason for hiding this comment

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

Some ideas for improvements to avoid future stumbles.

examples/hpc/dla-driver/src/tensor3.rs Outdated Show resolved Hide resolved
examples/hpc/dla-driver/src/tensor3.rs Outdated Show resolved Hide resolved
examples/hpc/dla-driver/src/tensor3.rs Outdated Show resolved Hide resolved
examples/hpc/dla-driver/src/tensor4.rs Show resolved Hide resolved
examples/hpc/dla-driver/src/tensor4.rs Outdated Show resolved Hide resolved
@vilukissa68
Copy link
Collaborator Author

Tested suggested changes with #51 and everything seems to work

@vilukissa68 vilukissa68 requested a review from hegza August 1, 2024 05:56
Copy link
Contributor

@hegza hegza left a comment

Choose a reason for hiding this comment

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

Ideally, replace unwrap with unwrap_unchecked where fitting but please merge afterwards.

examples/hpc/dla-driver/src/tensor3.rs Outdated Show resolved Hide resolved
@vilukissa68 vilukissa68 merged commit 38bc7d5 into soc-hub-fi:main Aug 1, 2024
6 checks passed
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.

2 participants