Skip to content

Incorrect Kind for bf16 #995

@laggui

Description

@laggui

Just stumbled upon this incorrect definition for bf16.

Shouldn't KIND be Kind::BFloat16 here?

tch-rs/src/wrappers/kind.rs

Lines 158 to 161 in ca61fe0

unsafe impl Element for half::bf16 {
const KIND: Kind = Kind::Half;
const ZERO: Self = half::bf16::ZERO;
}

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions