Skip to content

[FEA]: Support StridedMemoryView dlpack construction from kDLBfloat type code #1624

Description

@isVoid

Is this a duplicate?

Area

cuda.core

Is your feature request related to a problem? Please describe.

I would like StridedMemoryView.from_dlpack to be able to handle kDLBfloat data type, using ml_dtypes.bfloat as is underlying dtype medium.

Describe the solution you'd like

Adding handlers to kDLBfloats inside _memoryview.pyx::dtype_dlpack_to_numpy.

Describe alternatives you've considered

No response

Additional context

No response

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions