From 77ae744c181c02fade85b8b0536b5dba8928c461 Mon Sep 17 00:00:00 2001
From: mberenjk <146776561+mberenjk@users.noreply.github.com>
Date: Mon, 6 Jan 2025 14:05:38 -0600
Subject: [PATCH] removing FP8 product from allReduce test cases (#97)

* removing FP8 product from allReduce test cases

---------

Co-authored-by: Marzieh Berenjkoub <mberenjk@amd.com>
---
 src/all_reduce.cu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/all_reduce.cu b/src/all_reduce.cu
index 21e3ce3a9e..5302f86833 100644
--- a/src/all_reduce.cu
+++ b/src/all_reduce.cu
@@ -65,6 +65,8 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t
   ncclRedOp_t *run_ops;
   const char **run_typenames, **run_opnames;
   int type_count, op_count;
+  if((type == ncclFp8E4M3 || type == ncclFp8E5M2) && op == ncclProd)
+    return testSuccess;
 
   if ((int)type != -1) {
     type_count = 1;
@@ -88,6 +90,8 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t
 
   for (int i=0; i<type_count; i++) {
     for (int j=0; j<op_count; j++) {
+      if((i == ncclFp8E4M3 || i == ncclFp8E5M2) && j == ncclProd)
+        continue;
       TESTCHECK(TimeTest(args, run_types[i], run_typenames[i], run_ops[j], run_opnames[j], -1));
     }
   }