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
)
......
......@@ -5,4 +5,4 @@ func @conv_2d_nchw_fchw(%input: memref<?x?x?x?xf32>, %filter: memref<?x?x?x?xf32
ins(%input, %filter : memref<?x?x?x?xf32>, memref<?x?x?x?xf32>)
outs(%output : memref<?x?x?x?xf32>)
return
}
\ No newline at end of file
}
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
)
......
......@@ -5,4 +5,4 @@ func @conv_2d_nhwc_hwcf(%input: memref<?x?x?x?xf32>, %filter: memref<?x?x?x?xf32
ins(%input, %filter : memref<?x?x?x?xf32>, memref<?x?x?x?xf32>)
outs(%output : memref<?x?x?x?xf32>)
return
}
\ No newline at end of file
}
......@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment