Skip to content
Snippets Groups Projects
Commit 6007b60c authored by Joejiong's avatar Joejiong
Browse files

modify after review

parent bd87cdf8
No related branches found
No related tags found
No related merge requests found
set(BUDDY_CONV_OPT_ATTR avx512f)
set(LLVM_MLIR_BINARY_DIR ${BUDDY_CONV_OPT_BUILD_DIR}/../llvm/build/bin)
set(BUDDY_OPT_ATTR avx512f)
set(LLVM_MLIR_BINARY_DIR ${BUDDY_OPT_BUILD_DIR}/../llvm/build/bin)
add_custom_command(OUTPUT conv-2d-nchw-fchw.o
COMMAND ${LLVM_MLIR_BINARY_DIR}/mlir-opt ${BUDDY_SOURCE_DIR}/benchmarks/DeepLearning/Ops/Conv2DNchwFchwOp/Conv2DNchwFchw.mlir
......@@ -11,7 +11,7 @@ add_custom_command(OUTPUT conv-2d-nchw-fchw.o
-convert-std-to-llvm='emit-c-wrappers=1'
-reconcile-unrealized-casts|
${LLVM_MLIR_BINARY_DIR}/mlir-translate --mlir-to-llvmir |
${LLVM_MLIR_BINARY_DIR}/llc -mtriple=x86_64-unknown-linux-gnu -mattr=${BUDDY_CONV_OPT_ATTR}
${LLVM_MLIR_BINARY_DIR}/llc -mtriple=x86_64-unknown-linux-gnu -mattr=${BUDDY_OPT_ATTR}
--filetype=obj -o ${BUDDY_BINARY_DIR}/../benchmarks/DeepLearning/Ops/Conv2DNchwFchwOp/conv-2d-nchw-fchw.o
)
......
set(BUDDY_CONV_OPT_ATTR avx512f)
set(LLVM_MLIR_BINARY_DIR ${BUDDY_CONV_OPT_BUILD_DIR}/../llvm/build/bin)
set(BUDDY_OPT_ATTR avx512f)
set(LLVM_MLIR_BINARY_DIR ${BUDDY_OPT_BUILD_DIR}/../llvm/build/bin)
add_custom_command(OUTPUT conv-2d-nhwc-hwcf.o
COMMAND ${LLVM_MLIR_BINARY_DIR}/mlir-opt ${BUDDY_SOURCE_DIR}/benchmarks/DeepLearning/Ops/Conv2DNhwcHwcfOp/Conv2DNhwcHwcf.mlir
......@@ -13,7 +13,7 @@ add_custom_command(OUTPUT conv-2d-nhwc-hwcf.o
${LLVM_MLIR_BINARY_DIR}/mlir-translate --mlir-to-llvmir |
${LLVM_MLIR_BINARY_DIR}/llc
-mtriple=x86_64-unknown-linux-gnu
-mattr=${BUDDY_CONV_OPT_ATTR}
-mattr=${BUDDY_OPT_ATTR}
--filetype=obj
-o ${BUDDY_BINARY_DIR}/../benchmarks/DeepLearning/Ops/Conv2DNhwcHwcfOp/conv-2d-nhwc-hwcf.o
)
......
......@@ -14,7 +14,7 @@
//
//===----------------------------------------------------------------------===//
//
// This file implements the benchmark for depthwise conv2d(nhw) operation.
// This file implements the benchmark for conv2d(nhwc-hwcf) operation.
//
//===----------------------------------------------------------------------===//
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment