Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
wmsofts
Paddle
提交
c665400b
P
Paddle
项目概览
wmsofts
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
c665400b
编写于
3月 15, 2023
作者:
U
umiswing
提交者:
GitHub
3月 15, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[cutlass] Fix make (#51718)
上级
5cb95856
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
16 addition
and
5 deletion
+16
-5
cmake/external/cutlass.cmake
cmake/external/cutlass.cmake
+13
-5
paddle/phi/kernels/CMakeLists.txt
paddle/phi/kernels/CMakeLists.txt
+3
-0
未找到文件。
cmake/external/cutlass.cmake
浏览文件 @
c665400b
...
...
@@ -38,19 +38,27 @@ ExternalProject_Add(
PREFIX
${
CUTLASS_PREFIX_DIR
}
UPDATE_COMMAND
""
CONFIGURE_COMMAND
""
BUILD_COMMAND
BUILD_COMMAND
""
INSTALL_COMMAND
""
TEST_COMMAND
""
)
add_custom_target
(
cutlass_codegen
COMMAND
rm -rf
${
CMAKE_SOURCE_DIR
}
/paddle/phi/kernels/sparse/gpu/cutlass_generator/build &&
${
CMAKE_SOURCE_DIR
}
/paddle/phi/kernels/sparse/gpu/cutlass_generator/build
COMMAND
mkdir -p
${
CMAKE_SOURCE_DIR
}
/paddle/phi/kernels/sparse/gpu/cutlass_generator/build/generated/gemm
&&
${
PYTHON_EXECUTABLE
}
-B
COMMAND
${
PYTHON_EXECUTABLE
}
-B
${
CMAKE_SOURCE_DIR
}
/paddle/phi/kernels/sparse/gpu/cutlass_generator/gather_gemm_scatter_generator.py
"
${
THIRD_PARTY_PATH
}
/cutlass/src/extern_cutlass/tools/library/scripts/"
"
${
CMAKE_SOURCE_DIR
}
/paddle/phi/kernels/sparse/gpu/cutlass_generator/build"
"
${
CMAKE_CUDA_COMPILER_VERSION
}
"
INSTALL_COMMAND
""
TEST_COMMAND
""
)
VERBATIM
)
add_library
(
cutlass INTERFACE
)
add_dependencies
(
cutlass_codegen extern_cutlass
)
add_dependencies
(
cutlass extern_cutlass
)
paddle/phi/kernels/CMakeLists.txt
浏览文件 @
c665400b
...
...
@@ -174,6 +174,9 @@ set(ADD_PHI_KERNELS phi_cpu)
if
(
WITH_GPU OR WITH_ROCM
)
if
(
WITH_GPU
)
add_library
(
phi_gpu
${
kernel_cu
}
)
if
(
WITH_CUTLASS
)
add_dependencies
(
phi_gpu cutlass_codegen
)
endif
()
elseif
(
WITH_ROCM
)
hip_add_library
(
phi_gpu STATIC
${
kernel_cu
}
)
endif
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录