1 Star 0 Fork 1

jimmycao992/try_attention

forked from 王佰如/try_attention 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
CMakeLists.txt 1.08 KB
一键复制 编辑 原始数据 按行查看 历史
王佰如 提交于 2023-12-04 11:10 . try_attention泛化
cmake_minimum_required(VERSION 3.16)
# -------------------------------------------------------------------------------------------------------------------
# 1. ASCEND_INSTALL_PATH is your CANN package install path
set(ASCEND_INSTALL_PATH "/home/data/wbr/Ascend/ascend-toolkit/latest")
# -------------------------------------------------------------------------------------------------------------------
# 2. ASCEND_PRODUCT_TYPE should be in [ascend910, ascend310p or ascend910B1]
set(ASCEND_PRODUCT_TYPE ascend910B1)
# 3. ASCEND_CORE_TYPE should be in [AiCore VectorCore]
set(ASCEND_CORE_TYPE VectorCore)
# 4. ASCEND_RUN_MODE should be in [CPU SIMULATOR ONBOARD]"
# set(ASCEND_RUN_MODE CPU)
set(ASCEND_RUN_MODE ONBOARD)
# 5. Set operator name
set(smoke_testcase "mocha_energy")
set(CUSTOM_INCLUDE_PATH "${CMAKE_SOURCE_DIR}/include")
set(CMAKE_SYSTEM_NAME Linux)
set(CMAKE_CXX_STANDARD 17)
set(CCE_CMAKE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules)
list(APPEND CMAKE_MODULE_PATH ${CCE_CMAKE_PATH})
project(kernel_samples LANGUAGES CCE CXX)
add_subdirectory(cmake/cpu)
add_subdirectory(cmake/npu)
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/jimmycao992/try_attention.git
git@gitee.com:jimmycao992/try_attention.git
jimmycao992
try_attention
try_attention
master

搜索帮助