executorch icon indicating copy to clipboard operation
executorch copied to clipboard

Add permute operator to vulkan_partitioner.py

Open derekxu opened this issue 1 year ago • 2 comments

Summary: Add permute operator to vulkan_partitioner.py

Following the guide: https://docs.google.com/document/d/1xy4RoJLasHcrPL_lmwzgD3B2QaJ8Y8YdoZVILU6NZjc/edit

.cpp, glsl and yaml files of permute are implemented. 32 bit tests passed, 16bit tests skipped

cpp: https://www.internalfb.com/code/fbsource/xplat/executorch/backends/vulkan/runtime/graph/ops/impl/Permute.cpp glsl: https://www.internalfb.com/code/fbsource/xplat/executorch/backends/vulkan/runtime/graph/ops/glsl/permute.glsl

Differential Revision: D57113667

derekxu avatar May 08 '24 17:05 derekxu

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/3547

Note: Links to docs will display an error until the docs builds have been completed.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

pytorch-bot[bot] avatar May 08 '24 17:05 pytorch-bot[bot]

This pull request was exported from Phabricator. Differential Revision: D57113667

facebook-github-bot avatar May 08 '24 17:05 facebook-github-bot