pytorch/xla

[torch_xla] scan_layers fails if the layer have no weights

Open

#8,748 创建于 2025年2月26日

在 GitHub 查看
 (1 评论) (0 反应) (1 负责人)C++ (575 fork)github user discovery
enhancementgood first issue

仓库指标

Star
 (2,786 star)
PR 合并指标
 (PR 指标待抓取)

描述

Seen on https://github.com/pytorch/xla/pull/8734/files#diff-fd0b937d2fd231d227ae1901d62379ae35208c19b3e90f4188a1e272764eaae4

We should support scanning over layers that don't even have any weights and just do pure compute.

贡献者指南

研究方向
检查 torch xla 中的 scan layers 函数，了解它如何遍历层并处理没有权重的层。查看引用的 PR #8734 以获取上下文。修复应允许扫描没有权重的层，可能通过当权重列表为空时跳过与权重相关的操作。
技术栈
python
领域
machine learningbackend
议题类型
缺陷
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
半天
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
基本清晰
前置要求
PyTorchXLA basics
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
50