facebookresearch/xformers

Provide an easy way to debug mem-eff issues

开放

#632 创建于 2023年1月10日

 (0 条评论) (0 个反应) (0 位负责人)Python (473 个派生)batch import
good first issue

仓库指标

星标
 (6,903 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

🚀 Feature

We should be able to: (1) Understand which op is used for FW/BW (via additional logging enabled with an env variable?) (2) Possibly disable some ops entirely with env variable (like CUDA_VISIBLE_DEVICES but something like XFORMERS_MEMEFF_FW_OPERATORS=...)

贡献者指南