Skip to content

GitLab

Commit 8c9a37b1 authored by fengzch's avatar fengzch
Browse files

fix: compile dwconv.cu complete

parent b6b6fc4f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/kernels/dwconv.cu
View file @ 8c9a37b1
...@@ -259,7 +259,7 @@ Tensor dwconv_f16(Tensor input, Tensor weight, Tensor out, Tensor bias) { ...@@ -259,7 +259,7 @@ Tensor dwconv_f16(Tensor input, Tensor weight, Tensor out, Tensor bias) {
ThreadBlockOutputShape::kN, ThreadBlockOutputShape::kN,
ThreadBlockOutputShape::kH, ThreadBlockOutputShape::kH,
ThreadBlockOutputShape::kW>, ThreadBlockOutputShape::kW>,
4, 2,
cutlass::arch::OpMultiplyAdd, cutlass::arch::OpMultiplyAdd,
cutlass::conv::IteratorAlgorithm::kFixedStrideDilation, cutlass::conv::IteratorAlgorithm::kFixedStrideDilation,
cutlass::conv::StrideSupport::kFixed, cutlass::conv::StrideSupport::kFixed,
......
Markdown is supported
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