-
Notifications
You must be signed in to change notification settings - Fork 422
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
Optimized atan2, _softmax, cat, clamp, full, relu, remainder, permute_copy_out ops and updates to use memory_allocator #7567
base: main
Are you sure you want to change the base?
Conversation
Adding mean and where ops optimized on HiFi
* adding pow, remainder, minimum, maximum operators * adding pow, remainder, minimum, maximum operators
Adding quantized linear optimized versions for int8 and uint8
* Adding cat, full, permute_copy and relu ops (pytorch#34) * Adding cat, full, permute_copy * updating relu wrt new ref (pytorch#36) * Temporary memory allocation, replacing mallocs (pytorch#38) * Integrated temporary mem alloc functionality in place of malloc * Namespace related changes * Cleanup the main application * Adding atan2, softmax, clamp and remainder ops (pytorch#37) * Replaced malloc with temp_memory_allocator --------- Co-authored-by: nishpoonia <94543206+nishpoonia@users.noreply.github.com> Co-authored-by: Rushi-cad <gherderu@cadence.com>
* adding ET_KERNEL_CHECK for allocate_temp_memory * solving lint error * Removing redundant check
Adding _softmax, relu, permute etc
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/7567
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit fe5e7d7 with merge base 1bac885 (): This comment was automatically generated by Dr. CI and updates every 15 minutes. |
@pytorchbot label "topic: not user facing" |
- fixing build issue on previous commit
Update functions_hifi.yaml
…d removing exec_ten uses
Incorporating review comments: removing nesting to check data type an…
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.
Looks good from eyeballing, will link internally and solve any issues in follow up diffs
@@ -172,6 +179,7 @@ int main(int argc, char** argv) { | |||
|
|||
// Run the model. | |||
Error status = method->execute(); | |||
|
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.
please avoid empty line changes
"${EXECUTORCH_ROOT}/kernels/portable/cpu/op_clone.cpp" | ||
"${EXECUTORCH_ROOT}/kernels/portable/cpu/op_embedding.cpp" | ||
"${EXECUTORCH_ROOT}/kernels/portable/cpu/op_full.cpp" | ||
"${EXECUTORCH_ROOT}/kernels/portable/cpu/op_gt.cpp" |
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.
which op requires gt?
Summary
Optimized atan2, _softmax, cat, clamp, full, relu, remainder, permute_copy_out ops and updates to use memory_allocator
Test plan
Unit tested kernels