概述

kv_compress_epilog / pointnet_sa / rms_norm / voxel_conv 四个算子样例此前固定在旧版 DSL whl 上，无法在 CANNBot-DSL 0.6.0 上运行。本 MR 将这四个算子迁移至 0.6.0 新接口，并按代码检视意见完成整改，测试同步迁移。

改动范围仅限这四个算子（grouped_matmul / matmul 不在本次范围内，未做改动）。

接口迁移

• 模块路径迁至 cannbotdsl.ops.* / cannbotdsl.lang.*
• tile_view 更名 tile_slice；@jit 入口更名 @host
• 张量搬运改为 Channel.produce() / consume() 显式槽位协议
• torch 张量直传，去掉 as_tensor 转换层
• vf 寄存器模式 raw → simd；删除 vf 结尾不再需要的 membar
• pointnet_sa 按新接口重写 host 侧并统一 tiling 属性为 snake_case
• rms_norm 采用 arena 维护版内核与接口；voxel_conv 补齐 produce/consume 槽位协议与 @host 入口

代码检视整改

• host 侧不再校验 tensor 是否在 NPU；不再读取 device tensor 的值（移除 .cpu() / .item() 路径）
• workspace 与输出 tensor 统一用 torch.empty 构造
• 校验由 assert 改为显式 raise ValueError / TypeError，只校验算子规格
• 类名统一大驼峰，tiling 属性统一 snake_case
• 清理未使用的 import 与变量；修正首字母小写的 docstring
• metadata 与主算子 py 文件同目录

测试

四个算子的测试同步迁移至 @host 入口与新模块路径，61 个用例全部通过。

🤖 Generated with Claude Code