handle out != None

Author: danielvegamyhre

torchao/prototype/moe_training/scaled_grouped_mm.py

@@ -40,7 +40,11 @@ def _scaled_grouped_mm(
         offs (int32 torch.Tensor): The offsets to use to mark the starting index of each group along dim0 of the A tensor.
         out_dtype (Optional[torch.dtype]): The dtype of the output tensor. Currently only torch.bfloat16 is supported.
     """
+<<<<<<< HEAD
     # logger.info("Using scaled_grouped_mm")
+=======
+    #logger.info("Using scaled_grouped_mm")
+>>>>>>> 6ca070de (handle out != None)
     return _Float8GroupedMM.apply(
         A,
         B_t,