| feat(fast_kernel_launch_example): 在add_npu中添加 OptionalDeviceGuard Co-authored-by: chen-shuai<chenshuai96@huawei.com> # message auto-generated for no-merge-commit merge: !2053 merge optional_device_guard into master feat(fast_kernel_launch_example): 在add_npu中添加 OptionalDeviceGuard Created-by: chen-shuai Commit-by: chen-shuai Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 通过配置OptionalDeviceGuard,确保后续操作在正确的设备上下文中执行 作用说明: OptionalDeviceGuard 是一个设备上下文管理器 它以输入张量 x 的设备作为当前设备 确保 add_meta 和后续 NPU 内核执行在正确的设备上进行 RAII 风格:析构时自动恢复之前的设备状态 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1142 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 按照examples/fast_kernel_launch_example/README.md本地测试 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2053 | 4 个月前 |