add the missing pieces (#392)

Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
2021-12-25 04:29:54 -08:00
parent dceabd4c5a
commit f78994bb40
3 changed files with 8 additions and 0 deletions
@@ -41,6 +41,7 @@ class GemmOperation:
    complex_operators = [
      MathOperation.multiply_add_complex, 
      MathOperation.multiply_add_complex_gaussian,
+      MathOperation.multiply_add_complex_fast_f32
    ]
    return self.tile_description.math_instruction.math_operation in complex_operators