super_kernel样例使用指导

功能描述

使用super_kernel完成算子融合

目录结构

├── super_kernel_base             # 基础功能的样例
│  └── superkernel_scope.py       # 通过super_kernel完成算子融合
└── super_kernel_profiling        # 展示profiling的样例
  └── superkernel_compare.py      # 使用super_kernel与不使用super_kernel的数据进行对比
└── super_kernel_runtime_ascendc_only        # 极简super_kernel样例
   └── superkernel_runtime_ascendc_basic.py  # 通过ascendc编译super_kernel完成算子融合，并使用runtime运行时环境执行

前置说明

请务必参考《源码构建指南》完成前置环境准备。

依赖安装

样例执行所需的Python依赖已写入requirements.txt，可通过以下命令安装：

pip3 install -r requirements.txt

用例演示

用例1

用例2

用例3

参考

请参考《Ascend Extension for PyTorch》中“套件与三方库 > PyTorch图模式使用(TorchAir) > API参考 > torchair.scope > super_kernel”的相关内容。