This repository has been archived by the owner on Jan 24, 2024. It is now read-only.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
CINN基础算子代码开发示例。
仅供参考!!!算子本身无任何实际意义!!!
提示
CINN_REGISTER_HELPER
名称。bash build.sh gpu_on ci
命令来本地运行所有单测,或通过如下命令运行单个单测:部分文件作用说明
必备文件
cinn/hlir/op/contrib/example.cc:算子代码实现
cinn/hlir/op/use_ops.h & cinn/hlir/op/contrib/CMakeLists.txt:编译&链接所需
cinn/frontend/net_builder.h:对外API接口
cinn/pybind/frontend.cc:python接口
python/tests/ops/test_example_op.py:python测试文件
可选文件
cinn/runtime/cpu/host_intrinsics.cc:自定义函数host端实现及注册
cinn/runtime/cuda/cinn_cuda_runtime_source.cuh & cinn/runtime/cuda/cuda_intrinsics.cc:自定义函数nvgpu端实现及注册