1 В избранное 0 Ответвления 0

GITHUB-MIRROR/MooreThreads-torch_musa

 / Детали:

Поддержка MUSA bf16

Предстоит сделать
Владелец
Создано  
08.02.2025

MUSA поддерживает bf16?

import torch
import torch_musa

torch.empty((1,), device="musa:0", dtype=torch.bfloat16, memory_format=torch.contiguous_format)
RuntimeError: Поддерживаемый тип тензора не найден: c10::BFloat16

Чувствую, что делаю что-то не так...

Комментарий (0)

GitLife Service Account Задача создана

Вход Перед тем как оставить комментарий

Статус
Ответственный
Контрольная точка
Pull Requests
Связанные запросы на слияние могут быть закрыты после их объединения
Ветки
Дата начала   -   Крайний срок
-
Закрепить/Открепить
Приоритет
Участники(1)
1
https://api.gitlife.ru/github-mirror/MooreThreads-torch_musa.git
git@api.gitlife.ru:github-mirror/MooreThreads-torch_musa.git
github-mirror
MooreThreads-torch_musa
MooreThreads-torch_musa