feat: add args to allow user can control the communication datatype

2025-08-23 13:43:14 +00:00
parent b780a35577
commit 3a97a82f52
5 changed files with 88 additions and 33 deletions
@@ -394,9 +394,11 @@ extern "C" {

        // Tensor dumping path - if provided, network communication tensors will be dumped
        const char * dump_folder;
-
+        
        // Enable/disable communication and computation logging for gantt chart analysis
        bool enable_comm_compute_log;
+        
+        const char * comm_datatype;
    };

    // model quantization parameters