Auto buwbw cal

README.md

@@ -140,6 +140,11 @@ $ ./scripts/build.sh -c ns3
 $  ./bin/SimAI_analytical -w example/workload_analytical.txt -g 9216 -g_p_s 8 -r test- -busbw example/busbw.yaml
 ```
 
+For calculating bus bandwidth autolly, please try the following command:
+```bash
+$  ./bin/SimAI_analytical -w ./example/workload_analytical.txt -g 9216  -nv 360 -nic 48.5 -n_p_s 8 -g_p_s 8 -r example-
+```
+
 ## Use SimAI-Simulation
 
 ```bash

astra-sim-alibabacloud/astra-sim/network_frontend/analytical/AnalyticalAstra.cc

@@ -67,8 +67,8 @@ struct user_param {
 
 int main(int argc,char *argv[]) {
   UserParam* param = UserParam::getInstance();
-  if (param->parseArg(argc,argv)) {
-    std::cerr << "-h,       --help                Help message" << std::endl;
+  if (param->parse(argc,argv)) {
+    std::cerr << "-h,     --help              Help message" << std::endl;
     return -1;
   }
   param->mode = ModeType::ANALYTICAL;

astra-sim-alibabacloud/astra-sim/system/Common.hh

@@ -11,7 +11,7 @@ LICENSE file in the root directory of this source tree.
 #include <vector>
 #include "AstraNetworkAPI.hh"
 
-enum class GPUType { A100, A800, H100, H800, NONE };
+enum class GPUType { A100, A800, H100, H800, NONE, H20};
 
 namespace AstraSim {
 #define CLOCK_PERIOD 1

astra-sim-alibabacloud/astra-sim/system/Sys.cc

@@ -14,6 +14,7 @@ LICENSE file in the root directory of this source tree.
 #include "Common.hh"
 #include "RendezvousRecvData.hh"
 #include "RendezvousSendData.hh"
+#include "calbusbw.h"
 #include "astra-sim/system/collective/AllToAll.hh"
 #include "astra-sim/system/collective/DoubleBinaryTreeAllReduce.hh"
 #include "astra-sim/system/collective/HalvingDoubling.hh"
@@ -264,6 +265,11 @@ Sys::Sys(
     std::atexit(exiting);
     std::cout << "total nodes: " << total_nodes << std::endl;
   }
+  #ifdef ANALYTI
+  nic_ratio_data = readCSV(NIC_RATIO_PATH);
+  nvlink_ratio_data = readCSV(NVLINK_RATIO_PATH);
+  ata_ratio_data = readCSV(ATA_RATIO_PATH);
+  #endif
   NI->sim_init(MEM);
   memBus = new MemBus(
       "NPU",

astra-sim-alibabacloud/astra-sim/system/Sys.hh

@@ -148,7 +148,9 @@ class Sys : public Callable {
   int total_running_streams;
   std::map<int, std::list<BaseStream*>> active_Streams;
   std::map<int, std::list<int>> stream_priorities;
-
+  std::vector<std::vector<std::string>> nic_ratio_data;
+  std::vector<std::vector<std::string>> nvlink_ratio_data;
+  std::vector<std::vector<std::string>> ata_ratio_data;
   QueueLevels* vLevels;
   std::map<std::string, LogicalTopology*> logical_topologies;
   std::map<Tick, std::list<std::tuple<Callable*, EventType, CallData*>>>