Skip to content

Commit 4c9adc0

Browse files
committed
Fix hcdnnSetTensorNdDescriptor parameter order
Remove HCDNN_TENSOR_NCHW parameter from all three hcdnnSetTensorNdDescriptor calls. Unlike hcdnnSetTensor4dDescriptor, hcdnnSetTensorNdDescriptor follows cuDNN conventions and only takes (descriptor, dataType, nbDims, dims[], strides[]) without a format parameter.
1 parent 13316df commit 4c9adc0

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

src/infiniop/ops/gemm/metax/hcdnn/gemm_hcdnn.cc

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,6 @@ infiniStatus_t calculate(
9696
1 };
9797
CHECK_MCDNN(hcdnnSetTensorNdDescriptor(
9898
aDesc,
99-
HCDNN_TENSOR_NCHW,
10099
hcdnn_dtype,
101100
3,
102101
a_dims.data(),
@@ -121,7 +120,6 @@ infiniStatus_t calculate(
121120
1 };
122121
CHECK_MCDNN(hcdnnSetTensorNdDescriptor(
123122
bDesc,
124-
HCDNN_TENSOR_NCHW,
125123
hcdnn_dtype,
126124
3,
127125
b_dims.data(),
@@ -136,7 +134,6 @@ infiniStatus_t calculate(
136134
1 };
137135
CHECK_MCDNN(hcdnnSetTensorNdDescriptor(
138136
cDesc,
139-
HCDNN_TENSOR_NCHW,
140137
hcdnn_dtype,
141138
3,
142139
c_dims.data(),

0 commit comments

Comments
 (0)