/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/fx/profiler/experimental/profiler_function/python_ops.py
18 строк
402 B
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
import operator from typing import Any, Tuple from ..registry import meta_profiler_function @meta_profiler_function.register(operator.getitem) def operator_getitem(a: Any, b: Any) -> Tuple[int, int]: flops = 0 macs = 0 return flops, macs @meta_profiler_function.register(getattr) def python_getattr(a: Any, b: Any) -> Tuple[int, int]: flops = 0 macs = 0 return flops, macs