Skip to content

To do: Softmax masking should be optimized #84

@andrej

Description

@andrej

void mask_bf16(bfloat16 *inout, const int32 unmasked_size, const int32 total_size)
{
// TODO: Optimize this to use vector code
for (int32 i = unmasked_size; i < total_size; i++) {
inout[i] = (bfloat16)(-INFINITY);
}
}

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions