File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -224,7 +224,8 @@ struct ExampleRunner {
224
224
std::vector<ElementA> a (K * M * L);
225
225
std::vector<ElementB> b (K * N * L);
226
226
std::vector<ElementC> c (M * N * L);
227
- std::vector<ElementC> d (M * N * L, ElementC{0 });
227
+ std::vector<ElementC> d (M * N * L, ElementC{-1 });
228
+ std::vector<ElementC> ref_d (M * N * L, ElementC{-2 });
228
229
229
230
fill_matrix (a);
230
231
fill_matrix (b);
@@ -234,7 +235,7 @@ struct ExampleRunner {
234
235
block_B.copy_from_host (b.data (), b.size ());
235
236
block_C.copy_from_host (c.data (), c.size ());
236
237
block_D.copy_from_host (d.data (), d.size ());
237
- block_ref_D.copy_from_host (d .data (), d.size ());
238
+ block_ref_D.copy_from_host (ref_d .data (), d.size ());
238
239
}
239
240
240
241
virtual void run (const Options& options, const cutlass::KernelHardwareInfo& hw_info) {
You can’t perform that action at this time.
0 commit comments