Skip to content

random_() is not supported for bfloat16 CUDA tensors on Windows #33793

Closed
@pbelevich

Description

@pbelevich

This is a known issue, that requires further investigation, currently calling any random_() method on bfloat16 CUDA tensor on Windows makes CUDA context invalid, all subsequent CUDA calls fail with 'CUDA error: unspecified launch failure'

Assignee, please look for "TODO: https://github.com/pytorch/pytorch/issues/33793" in the source code

cc @ngimel

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: bfloat16module: cudaRelated to torch.cuda, and CUDA support in generalmodule: randomRelated to random number generation in PyTorch (rng generator)triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions