AscendC样例指定npu
收藏回复举报
AscendC样例指定npu
t('forum.solved') 已解决
发表于2024-12-17 17:23:32
0 查看

如何指定npu来运行脚本呢?

目前的样例给的命令是:

bash run.sh -r cpu -v Ascend310P1

在多卡环境下是否可以指定npu呢?

仓库链接:

operator/AddCustomSample/KernelLaunch/AddKernelInvocation · Ascend/samples - 码云 - 开源中国

我要发帖子