Skip to content

Commit e9dc0f1

Browse files
author
Timmy
committed
undate
1 parent 15c03be commit e9dc0f1

File tree

1 file changed

+17
-3
lines changed

1 file changed

+17
-3
lines changed

src/client/clfunc_xgemm.hpp

Lines changed: 17 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -522,9 +522,23 @@ class xGemm : public clblasFunc
522522
err = clEnqueueWriteBufferRect(queue_, buffer_.buf_c_, CL_TRUE, c_buffer_origin, c_host_origin, c_region, c_buffer_row_pitch,
523523
c_buffer_slice_pitch, c_host_row_pitch, c_host_slice_pitch, buffer_.c_, 0, NULL, NULL);
524524

525-
buffer_.lda_ = 0;
526-
buffer_.ldb_ = 0;
527-
buffer_.ldc_ = 0;
525+
if(buffer_.trans_a_==clblasNoTrans)
526+
{
527+
buffer_.lda_=buffer_.m_;
528+
}
529+
else
530+
{
531+
buffer_.lda_=buffer_.k_;
532+
}
533+
if(buffer_.trans_b_==clblasNoTrans)
534+
{
535+
buffer_.ldb_=buffer_.k_;
536+
}
537+
else
538+
{
539+
buffer_.ldb_=buffer_.m_;
540+
}
541+
buffer_.ldc_=buffer_.m_;
528542
xGemm_Function(false);
529543
/*
530544
err = clEnqueueReadBuffer(queue_, buffer_.buf_c_, CL_TRUE,

0 commit comments

Comments
 (0)