Skip to content

[Vulkan] Make partitioning dtype aware #13084

@SS-JIA

Description

@SS-JIA

🚀 The feature, motivation and pitch

Context

Currently, Vulkan's partitioner does not account for the dtype of expected input/output tensors participating in an op. This can lead to a situation where the runtime receives dtypes for an op that it can't handle.

Alternatives

No response

Additional context

No response

RFC (Optional)

No response

cc @manuelcandales @cbilgin

Metadata

Metadata

Assignees

No one assigned

    Labels

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

    Type

    No type

    Projects

    Status

    Ready

    Milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions