From f9ece1b42c62899017dae5340b40e543d0a5b36b Mon Sep 17 00:00:00 2001 From: jeromeku Date: Sat, 30 Mar 2024 06:00:46 -0700 Subject: [PATCH] Python `Gemm` `tile_descriptions` fix (#1439) * fix python gemm tile descriptions * fix formatting * fix math_operation filtering * fix formatting --- python/cutlass/op/gemm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cutlass/op/gemm.py b/python/cutlass/op/gemm.py index c78033bd..e486a431 100644 --- a/python/cutlass/op/gemm.py +++ b/python/cutlass/op/gemm.py @@ -403,7 +403,7 @@ class Gemm(OperationBase): """ tds = [datatypes.td_from_profiler_op(op) for op in self.possible_operations.all_operations] 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 def construct( @@ -712,4 +712,4 @@ class Gemm(OperationBase): if sync: arguments.sync() - return arguments + return arguments \ No newline at end of file