- my_add: c = 2 * a +b
- my_matmul: c = a * b，使用cublas库实现矩阵乘
python3 setup.py install# 测试my_add并导出为onnx
python3 my_add_op_test.py
# 测试my_matmul并导出为onnx
python3 my_matmul_op_test.py -
Notifications
You must be signed in to change notification settings - Fork 0
zhaocc1106/torch_custom_op
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
实现自己的torch算子
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published