Skip to content

Add support for torch.complex64 on vulkan backend #17791

@msluszniak

Description

@msluszniak

🚀 The feature, motivation and pitch

When trying to export whisper on vulkan backend in decoder export I got the error:

AssertionError: Invalid dtype for vulkan_preprocess (torch.complex64)

This is limiting my ability to export the decoder part. This will be also beneficial for exporting other models as well.

Alternatives

No alternatives that I'm aware of.

Additional context

No response

RFC (Optional)

No response

cc @SS-JIA @manuelcandales @digantdesai @cbilgin

Metadata

Metadata

Assignees

Labels

module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions