-
Notifications
You must be signed in to change notification settings - Fork 685
Open
Labels
module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/Issues related to the Vulkan delegate and code under backends/vulkan/
Milestone
Description
🐛 Describe the bug
Context
During export, the Vulkan delegate will downcast int64 and float64 tensors to int32 and int32 since 64-bit types are cumbersome to handle in compute shaders. However, this produces an issue the inputs and outputs of a model are 64-bit types; the ExecuTorch runtime will still be expecting to receive 64-bit data, but under the hood Vulkan will be treating the inputs and outputs as 32-bit.
Fixing
- Add a mechanism to allow the Vulkan delegate to automatically convert data to/from the source/destination ETensor type during input and output.
Versions
N/A
Metadata
Metadata
Assignees
Labels
module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/Issues related to the Vulkan delegate and code under backends/vulkan/
Type
Projects
Status
Ready