File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
src/library/blas/AutoGemm Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -351,11 +351,11 @@ def makeOpenCLKernelString(kernel):
351
351
kStr += endLine
352
352
kStr += " /* load global -> local */" + endLine
353
353
numALoads = (kernel .workGroupNumRows * kernel .microTileNumRows * kernel .unroll ) \
354
- / (kernel .workGroupNumRows * kernel .workGroupNumCols )
354
+ // (kernel .workGroupNumRows * kernel .workGroupNumCols ) # // -- integer divide
355
355
numALoadsR = (kernel .workGroupNumRows * kernel .microTileNumRows * kernel .unroll ) \
356
356
% (kernel .workGroupNumRows * kernel .workGroupNumCols )
357
357
numBLoads = (kernel .workGroupNumCols * kernel .microTileNumCols * kernel .unroll ) \
358
- / (kernel .workGroupNumRows * kernel .workGroupNumCols )
358
+ // (kernel .workGroupNumRows * kernel .workGroupNumCols ) # // - integer divide
359
359
numBLoadsR = (kernel .workGroupNumCols * kernel .microTileNumCols * kernel .unroll ) \
360
360
% (kernel .workGroupNumRows * kernel .workGroupNumCols )
361
361
Original file line number Diff line number Diff line change @@ -89,11 +89,11 @@ def isValid(self):
89
89
return True
90
90
"""
91
91
numALoads = (self.workGroupNumRows*self.microTileNumRows*self.unroll) \
92
- / (self.workGroupNumRows*self.workGroupNumCols)
92
+ // (self.workGroupNumRows*self.workGroupNumCols)
93
93
numALoadsR = (self.workGroupNumRows*self.microTileNumRows*self.unroll) \
94
94
% (self.workGroupNumRows*self.workGroupNumCols)
95
95
numBLoads = (self.workGroupNumCols*self.microTileNumCols*self.unroll) \
96
- / (self.workGroupNumRows*self.workGroupNumCols)
96
+ // (self.workGroupNumRows*self.workGroupNumCols)
97
97
numBLoadsR = (self.workGroupNumCols*self.microTileNumCols*self.unroll) \
98
98
% (self.workGroupNumRows*self.workGroupNumCols)
99
99
if (numALoads>0 and numALoadsR>0):
You can’t perform that action at this time.
0 commit comments