Fix a missing attribute issue with FP8 rowwise gemm (#52)

Summary: Pull Request resolved: #52

Reviewed By: xuzhao9

Differential Revision: D65950130

Pulled By: htyu

fbshipit-source-id: 43ba93eab73cd05a55ce05d5c605de23d688c73f

Author: htyu

tritonbench/operators/fp8_gemm_rowwise/operator.py

@@ -117,7 +117,7 @@ def __init__(
         super().__init__(tb_args, extra_args)
         self.use_cuda_graphs = True
         addmm_args = parse_args(self.extra_args)
-        if tb_args.production_shapes:
+        if hasattr(tb_args, "production_shapes") and tb_args.production_shapes:
             self.shapes = get_production_shapes(self.name, "fp8_gemm")
         elif addmm_args.m and addmm_args.n and addmm_args.k:
             self.shapes = [(addmm_args.m, addmm_args.n, addmm_args.k)]