昇腾910C爆显存,三步查清算子对齐
PyTorch 模型迁到 CANN 后一跑就 OOM,先把加卡放到最后。固定输入复现峰值,用 Profiler 对齐自定义算子前后的 shape、dtype、format 与 Data Copy,再根据证据处理算子融合和
阅读全文关键词:CANN Profiler
PyTorch 模型迁到 CANN 后一跑就 OOM,先把加卡放到最后。固定输入复现峰值,用 Profiler 对齐自定义算子前后的 shape、dtype、format 与 Data Copy,再根据证据处理算子融合和
阅读全文