Skip to content

❓ [Question] Why is torch.bfloat16 excluded from the allowed_casts set ? #3637

Open
@junstar92

Description

@junstar92

allowed_casts = {
torch.float,
torch.int32,
torch.int64,
torch.bool,
torch.int8,
torch.float16,
}

Is there a specific reason why torch.bfloat16 is not included in the allowed_casts set within the to_copy_dtype_validator function?

Plus, this causes graph partitioning when performing a aten.ops._to_copy operation to torch.bfloat16. I'm wondering if this could potentially impact performance.

Metadata

Metadata

Assignees

No one assigned

    Labels

    questionFurther information is requested

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions