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

OSCHINA-MIRROR/mindspore-community

 / Детали:

Интерфейсы PyBoost и адаптация оператора aclnnLeScalar ACLNN для работы с le

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

Существующий линейный оператор подключен к aclnnLeScalar, что позволяет избежать преобразования типа при использовании other как скалара, тем самым повышая производительность оператора.

Задачи

le

Номер Задача Подзадача Статус (новый/измененный/нет изменения/не применимо) Примечание
1 Базовая функциональность интерфейса Primitive Новый Добавлено LessEqualScalar
Functional Изменено Переопределены less_equal_scalar_op.yaml и less_equal_op.yaml
Tensor Изменено Переопределены less_equal_scalar_op.yaml и less_equal_op.yaml, CPU/GPU и статический граф проходят через py_method
2 Поддержка backend и типов данных Ascend Без изменений
GPU Без изменений
CPU Без изменений
3 Поддержка vmap Без изменений
4 Поддержка динамического Shape Динамическое Shape Без изменений
Динамическая ранг Без изменений
5 Поддержка обратного распространения Bprop функция Без изменений
Поддержка комплексных чисел Без изменений
6 Улучшение документации API отображение Без изменений
Документация интерфейсов на английском и китайском Изменено Изменена документация функционального интерфейса и тензоров на английском и китайском
7 Оптимизация производительности CPU Без изменений
GPU Не применимо

Общий фон

1. Сравнение с аналогами

  • Ссылка на аналогичный интерфейс:
    torch.le

  • Поддерживаемые типы данных аналога: INT8, UINT8, INT16, INT32, INT64, FLOAT16, FLOAT32, FLOAT64, BFLOAT16, BOOL

2. Ситуация с операторами в MindSpore

  • Текущие поддерживаемые типы данных
Ascend: INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT16, FLOAT, DOUBLE, BFLOAT16 (только для 910b), BOOL
  • Поддержка операторов после объединения трех backends (аналог + объединение всех трех backends)
    INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT16, FLOAT, DOUBLE, BFLOAT16 (только для 910b), BOOL

Введение

1. Описание функциональности

Элемент за элементом сравнивает значения входных параметров input <= other, выводит тип данных Tensor типа bool.

2. Описание интерфейса (mint/tensor/nn)

less_equal:
  - op_yaml: less_equal_scalar_op.yaml
    py_method: less_greater_equal
    Ascend: pyboost
    CPU: py_method
    GPU: py_method
    interface: tensor, function

  - op_yaml: less_equal_op.yaml
    py_method: less_greater_equal
    Ascend: pyboost
    CPU: pyboost
    GPU: pyboost
    interface: tensor, function

  - op_yaml: deprecated/less_equal_method.yaml
    py_method: deprecated_tensor_less_equal
    Ascend: py_method
    CPU: py_method
    GPU: py_method
    interface: tensor

x.less_equal(other)
le:
  alias: less_equal

x.le(other)
  • Операционный примитив
    mindspore/python/mindspore/ops/auto_generate/gen_ops_prim.py
    class LessEqualScalar(Primitive):
    Параметр Тип Вход/Выход/Атрибут Описание
    input Tensor Вход
    other Скаляр Вход
    output Tensor Выход

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

GitLife Service Account Задача создана
GitLife Service Account добавлено
 
v2.1.0
label.
GitLife Service Account добавлено
 
sig/ops
label.
GitLife Service Account добавлено
 
stable
label.
GitLife Service Account добавлено
 
迭代6
label.
Развернуть журнал операций

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

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