Skip to main content

1 功能描述

由于在ascendc算子开发过程中运行算子比较复杂,为了简化算子的运行,将运行算子变成可以用python直接调用的函数。所以编写了此代码。

2 安装

  • 请确保安装了 cann toolkit 与 官方算子包。
pip install l0n0lacl
source /home/HwHiAiUser/Ascend/ascend-toolkit/set_env.sh

3 通过创建算子库来运行算子

3.1 通过命令创建算子库

3.1.1 命令介绍

l0n0lcreateopps -h
usage: l0n0lcreateopps [-h] [--跳过自定义算子] 目标目录

为所有算子创建函数:生成算子代码并支持跳过自定义算子

positional arguments:
  目标目录           指定生成算子的目标目录路径(例如:'./operators/')

options:
  -h, --help     show this help message and exit
  --跳过自定义算子, -s  若启用,跳过

3.1.2 创建算子库

l0n0lcreateopps ./opp

执行完毕后 opp就成为了算子库

3.2 调用算子实例

from l0n0lacl import *
import numpy as np
import opp
a = 张量(np.ones((2, 3)) * 2)
other =  张量(np.ones((2, 3)) * 3)
out = 张量(np.zeros([2, 3]))
opp.Add(a, other, 1.2, out)
print(out)
[[5.6 5.6 5.6]
 [5.6 5.6 5.6]]

4 简单运行算子实例

4.1 先切换到cann环境,比如我的环境是:

source /home/HwHiAiUser/Ascend/ascend-toolkit/set_env.sh

4.2 设置要用到的设备

export ASCEND_VISIBLE_DEVICES=0,1

4.3 运行算子

from l0n0lacl import *
import numpy as np
import ctypes
import time
设备.设置内存共享组([0, 1])
a = np.random.uniform(-2, -1, (1024))
a_out = np.zeros_like(a)
a = 张量(a)
a_out = 张量(a_out)
a.切换到设备(1)
# a_out.切换到设备(1)
fn = 算子运行器('Abs')
out = fn(a, a_out)
print(out[1])

fn = 算子运行器('InplaceAcos')
a = np.random.uniform(-1, 1, (2000,2000)).astype(np.float16)
print(a)
out = fn(a)
print(out[0])

fn = 算子运行器('AdaptiveAvgPool2d')
a = np.random.uniform(0, 100, (2, 100, 100)).astype(np.float32)
out = np.zeros((2, 3, 3), dtype=a.dtype)
a = 张量(a, 格式=张量格式.NCL)
out = 张量(out).变更格式(张量格式.NCL)
output = fn(a, [3, 3], out)
print(output[2])


fn = 算子运行器('Addmv')
s = np.ones(3, dtype=np.float32)
mat = np.random.uniform(-1, 1, (3, 40000)).astype(np.float32)
vec = np.random.uniform(-1, 1, 40000).astype(np.float32)
alpha = 1.2
beta = 标量(1.1)
out = np.zeros(3, dtype=np.float32)
output = fn(s, mat, vec, alpha, beta, out, ctypes.c_int8(1))
print(output[-2])

fn = 算子运行器('Any')
s = np.random.uniform(-1, -0.5, (3, 4))
out = np.zeros(3, dtype=np.bool_)
output = fn(s, [1], ctypes.c_bool(False), out)
print(output[-1])

4.3 算子查找顺序

如果 ${NO_VENDORS_OPP} != '1':
    查找 ${ASCEND_OPP_PATH}/vendors目录(自己写的算子默认安装目录) 
查找 ${ASCEND_HOME_PATH}/lib64/libopapi.so 支持的算子(也就是官方算子包)
  • NO_VENDORS_OPP 如果不需要使用自定义算子, 可以添加此环境变量
  • ASCEND_OPP_PATH cann自带环境变量 在(source /home/HwHiAiUser/Ascend/ascend-toolkit/set_env.sh)时设置
  • ASCEND_HOME_PATHcann自带环境变量在(source /home/HwHiAiUser/Ascend/ascend-toolkit/set_env.sh)时设置

Metadata

Release files for l0n0lacl 2.4.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for l0n0lacl 2.4.0
File Size Uploaded Provenance
l0n0lacl-2.4.0.tar.gz 22.7 kB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for l0n0lacl 2.4.0
File Interpreter ABI Platform
l0n0lacl-2.4.0-py3-none-any.whl Python 3 none any Details

Total release size: 48.8 kB

Release files / l0n0lacl-2.4.0.tar.gz

Download URL l0n0lacl-2.4.0.tar.gz
Size 22.7 kB
Tags Source
SHA-256 checksum
How to use checksums
a878d8b20dda439742b92d2d027c7e57a6d236d4ed5056b3f881b97b4e9d26e3
BLAKE2b-256 checksum
How to use checksums
8791cf912697baa3a5df870ef7d1f82bf733358129113a8b59ccc1420e5ab50a
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.14.6

Release files / l0n0lacl-2.4.0-py3-none-any.whl

Download URL l0n0lacl-2.4.0-py3-none-any.whl
Size 26.1 kB
Tags Python 3
SHA-256 checksum
How to use checksums
0f440c571446e628c1b6f10e490119a98c2281a518ab6cc0200e8f8f3ccc3a06
BLAKE2b-256 checksum
How to use checksums
5f3354851b0df496576e4f4c95447db2d8a065277b830c495d9249fb6481973e
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.14.6

Security for l0n0lacl 2.4.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page