squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Python Gemm tile_descriptions fix (#1439)

Browse Source 
* fix python gemm tile descriptions

* fix formatting

* fix math_operation filtering

* fix formatting
This commit is contained in:
jeromeku 2024-03-30 06:00:46 -07:00 committed by GitHub
parent 28cbacbf64
commit f9ece1b42c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/cutlass/op/gemm.py
View File

@ -403,7 +403,7 @@ class Gemm(OperationBase):
""" """
tds = [datatypes.td_from_profiler_op(op) for op in self.possible_operations.all_operations] tds = [datatypes.td_from_profiler_op(op) for op in self.possible_operations.all_operations]
if self._math_operation is not None: if self._math_operation is not None:
tds = [td for td in tds if td.tile_description.math_instruction == self._math_operation] tds = [td for td in tds if td.math_instruction.math_operation == self._math_operation]
return tds return tds
def construct( def construct(
@ -712,4 +712,4 @@ class Gemm(OperationBase):
if sync: if sync:
arguments.sync() arguments.sync()
return arguments return arguments