gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

GEMM: swapped loops in rC[m][n]*=alpha

Browse Source
This commit is contained in:
Philippe Tillet
2015-07-16 10:40:38 -04:00
parent 4b004e1cd5
commit 9de87da993
2 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/backend/templates/gemm.cpp
View File

@@ -149,7 +149,7 @@ gemm_parameters::gemm_parameters(unsigned int simd_width
stream.inc_tab();
///Declare
stream << sdtype << " rC[" << p_.mS << "][" << p_.nS << "];" << std::endl;
stream << sdtype << " rC[" << p_.mS << "][" << p_.nS << "] = {{0}};" << std::endl;
stream << vdtype << " rA[" << p_.kS << "][" << p_.mS/p_.simd_width << "];" << std::endl;
stream << vdtype << " rB[" << p_.kS << "][" << p_.nS/p_.simd_width << "];" << std::endl;
@@ -164,10 +164,6 @@ gemm_parameters::gemm_parameters(unsigned int simd_width
stream << "__global " << sdtype << "* Ai[" << npA << "];" << std::endl;
stream << "__global " << sdtype << "* Bi[" << npB << "];" << std::endl;
for(int_t m=0; m < p_.mS; ++m)
for(int_t n=0; n < p_.nS; ++n)
stream << "rC[" << m << "][" << n << "] = 0;" << std::endl;
stream << "A += offa;" << std::endl;
stream << "B += offb;" << std::endl;
stream << "C += offc;" << std::endl;
@@ -389,9 +385,9 @@ gemm_parameters::gemm_parameters(unsigned int simd_width
stream << "//Write back C" << std::endl;
//alpha
for(int_t n=0; n < p_.nS; ++n)
for(int_t m=0; m < p_.mS; ++m)
for(int_t n=0; n < p_.nS; ++n)
stream << "rC[" << m << "][" << n << "] *= alpha;" << std::endl;
stream << "rC[" << m << "][" << n << "] *= alpha;" << std::endl;
//beta
unsigned int ministartstride0 = p_.simd_width;