├── .gitignore ├── Dockerfile ├── License ├── README.md ├── aicb.py ├── core ├── __init__.py └── grouped_gemm_util.py ├── download └── AICB_v1.0.deb ├── images ├── detail_log.png ├── readme_01.png ├── result_log.png ├── simai_dingtalk.jpg ├── simai_wechat.jpeg ├── time_log.png ├── tutorial_1.png ├── tutorial_2.png ├── tutorial_3.png ├── tutorial_4.png ├── tutorial_5.png ├── tutorial_6.png └── tutorial_7.png ├── log_analyzer ├── __init__.py ├── analyze_res_csv.py ├── ds_comm_log_analyzer.py ├── log.py ├── plot.py └── utils.py ├── requirements.txt ├── results └── visual_output │ └── A100_example.html ├── run_suites.py ├── scripts ├── coll_comm_check.sh ├── deepspeed_llama.sh ├── inference_configs │ ├── deepseek_default.json │ ├── deepseek_default_aiob.json │ ├── qwen3_default.json │ └── qwen3_default_aiob.json ├── inference_workload_with_aiob.sh ├── megatron_gpt.sh ├── megatron_workload_with_aiob.sh └── run_in_cluster.py ├── training └── tutorial.md ├── utils ├── benchmark_logger.py ├── deepgemm_utils.py ├── timer.py └── utils.py ├── visualize ├── __init__.py ├── example.html ├── generate.py └── inputs │ └── A100_example.csv ├── workload ├── Workload_spec_v1.1.csv ├── aiob_inputs │ └── Example.txt ├── physical │ ├── micro_test │ │ ├── all_gather_workload.csv │ │ ├── all_reduce_workload.csv │ │ ├── all_to_all_workload.csv │ │ ├── multi_all_reduce_workload.csv │ │ └── reduce_scatter_workload.csv │ └── model_workload │ │ ├── G13B-M1-C01_GPT13B_megatron_tp8_pp1_mbs1.csv │ │ ├── G13B-M1-C02_GPT13B_megatron_tp8_pp1_mbs1_sp.csv │ │ ├── G175B-M1-C03_GPT175B_megatron_tp8_pp16_mbs1.csv │ │ ├── L13B-D1-C03_Llama13B_zero2_mbs1.csv │ │ ├── L13B-D1-C04_Llama13B_zero3_mbs1.csv │ │ ├── L30B-D1-C05_Llama30B_zero2_mbs1.csv │ │ ├── L30B-D1-C06_Llama30B_zero3_mbs1.csv │ │ ├── L65B-D1-C07_Llama65B_zero2_mbs1.csv │ │ ├── L65B-D1-C07_Llama65B_zero3_mbs1.csv │ │ ├── L65B-M1-C05_Llama65B_megatron_tp2_pp8_mbs1.csv │ │ ├── L7B-D1-C01_Llama7B_zero2_mbs1.csv │ │ ├── L7B-D1-C02_Llama7B_zero3_mbs1.csv │ │ └── L7B-M1-C04_Llama7B_megatron_tp2_pp1_mbs1.csv └── simAI │ ├── micro_test │ ├── all_gather.txt │ ├── all_reduce.txt │ ├── all_to_all.txt │ └── muti_all_reduce.txt │ └── model_workload │ ├── G13B-M1-C01_GPT13B_megatron_tp8_pp1_mbs1_A100.txt │ ├── G13B-M1-C02_GPT13B_megatron_tp8_pp1_mbs1_sp_A100.txt │ ├── G175B-M1-C03_GPT175B_megatron_tp8_pp1_mbs1_A100.txt │ ├── L65B-M1-C05_Llama65B_megatron_tp8_pp1_mbs1_A100.txt │ ├── L65B_D1_C08_Llama65B_deepspeed_zero3_A100.txt │ ├── L7B-D1-C02_Llama7B_deepspeed_zero3_A100.txt │ └── L7B-M1-C04_Llama7B_megatron_tp2_pp1_mbs1_A100.txt ├── workload_applyer.py └── workload_generator ├── SimAI_inference_workload_generator.py ├── SimAI_training_workload_generator.py ├── __init__.py ├── analysis_pytorch_trace.py ├── generate_collective_test.py ├── generate_deepspeed_stage1_2_workload.py ├── generate_deepspeed_stage3_workload.py ├── generate_ds_trace_replay_workload.py ├── generate_megatron_workload.py ├── mocked_model ├── MockedModel.py ├── __init__.py ├── inference │ ├── AiobDeepSeek.py │ ├── AiobQwen3.py │ ├── MockedDeepSeek.py │ ├── MockedQwen3.py │ └── __init__.py └── training │ ├── AiobDeepSeek.py │ ├── AiobMegatron.py │ ├── MockedDeepSeek.py │ ├── MockedDeepspeed.py │ ├── MockedMegatron.py │ └── __init__.py └── workload_generator.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/Dockerfile -------------------------------------------------------------------------------- /License: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/License -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/README.md -------------------------------------------------------------------------------- /aicb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/aicb.py -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/grouped_gemm_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/core/grouped_gemm_util.py -------------------------------------------------------------------------------- /download/AICB_v1.0.deb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/download/AICB_v1.0.deb -------------------------------------------------------------------------------- /images/detail_log.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/detail_log.png -------------------------------------------------------------------------------- /images/readme_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/readme_01.png -------------------------------------------------------------------------------- /images/result_log.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/result_log.png -------------------------------------------------------------------------------- /images/simai_dingtalk.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/simai_dingtalk.jpg -------------------------------------------------------------------------------- /images/simai_wechat.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/simai_wechat.jpeg -------------------------------------------------------------------------------- /images/time_log.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/time_log.png -------------------------------------------------------------------------------- /images/tutorial_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/tutorial_1.png -------------------------------------------------------------------------------- /images/tutorial_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/tutorial_2.png -------------------------------------------------------------------------------- /images/tutorial_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/tutorial_3.png -------------------------------------------------------------------------------- /images/tutorial_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/tutorial_4.png -------------------------------------------------------------------------------- /images/tutorial_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/tutorial_5.png -------------------------------------------------------------------------------- /images/tutorial_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/tutorial_6.png -------------------------------------------------------------------------------- /images/tutorial_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/images/tutorial_7.png -------------------------------------------------------------------------------- /log_analyzer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /log_analyzer/analyze_res_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/log_analyzer/analyze_res_csv.py -------------------------------------------------------------------------------- /log_analyzer/ds_comm_log_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/log_analyzer/ds_comm_log_analyzer.py -------------------------------------------------------------------------------- /log_analyzer/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/log_analyzer/log.py -------------------------------------------------------------------------------- /log_analyzer/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/log_analyzer/plot.py -------------------------------------------------------------------------------- /log_analyzer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/log_analyzer/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/visual_output/A100_example.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/results/visual_output/A100_example.html -------------------------------------------------------------------------------- /run_suites.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/run_suites.py -------------------------------------------------------------------------------- /scripts/coll_comm_check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/coll_comm_check.sh -------------------------------------------------------------------------------- /scripts/deepspeed_llama.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/deepspeed_llama.sh -------------------------------------------------------------------------------- /scripts/inference_configs/deepseek_default.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/inference_configs/deepseek_default.json -------------------------------------------------------------------------------- /scripts/inference_configs/deepseek_default_aiob.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/inference_configs/deepseek_default_aiob.json -------------------------------------------------------------------------------- /scripts/inference_configs/qwen3_default.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/inference_configs/qwen3_default.json -------------------------------------------------------------------------------- /scripts/inference_configs/qwen3_default_aiob.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/inference_configs/qwen3_default_aiob.json -------------------------------------------------------------------------------- /scripts/inference_workload_with_aiob.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/inference_workload_with_aiob.sh -------------------------------------------------------------------------------- /scripts/megatron_gpt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/megatron_gpt.sh -------------------------------------------------------------------------------- /scripts/megatron_workload_with_aiob.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/megatron_workload_with_aiob.sh -------------------------------------------------------------------------------- /scripts/run_in_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/scripts/run_in_cluster.py -------------------------------------------------------------------------------- /training/tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/training/tutorial.md -------------------------------------------------------------------------------- /utils/benchmark_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/utils/benchmark_logger.py -------------------------------------------------------------------------------- /utils/deepgemm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/utils/deepgemm_utils.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/utils/timer.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/utils/utils.py -------------------------------------------------------------------------------- /visualize/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /visualize/example.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/visualize/example.html -------------------------------------------------------------------------------- /visualize/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/visualize/generate.py -------------------------------------------------------------------------------- /visualize/inputs/A100_example.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/visualize/inputs/A100_example.csv -------------------------------------------------------------------------------- /workload/Workload_spec_v1.1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/Workload_spec_v1.1.csv -------------------------------------------------------------------------------- /workload/aiob_inputs/Example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/aiob_inputs/Example.txt -------------------------------------------------------------------------------- /workload/physical/micro_test/all_gather_workload.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/micro_test/all_gather_workload.csv -------------------------------------------------------------------------------- /workload/physical/micro_test/all_reduce_workload.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/micro_test/all_reduce_workload.csv -------------------------------------------------------------------------------- /workload/physical/micro_test/all_to_all_workload.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/micro_test/all_to_all_workload.csv -------------------------------------------------------------------------------- /workload/physical/micro_test/multi_all_reduce_workload.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/micro_test/multi_all_reduce_workload.csv -------------------------------------------------------------------------------- /workload/physical/micro_test/reduce_scatter_workload.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/micro_test/reduce_scatter_workload.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/G13B-M1-C01_GPT13B_megatron_tp8_pp1_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/G13B-M1-C01_GPT13B_megatron_tp8_pp1_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/G13B-M1-C02_GPT13B_megatron_tp8_pp1_mbs1_sp.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/G13B-M1-C02_GPT13B_megatron_tp8_pp1_mbs1_sp.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/G175B-M1-C03_GPT175B_megatron_tp8_pp16_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/G175B-M1-C03_GPT175B_megatron_tp8_pp16_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L13B-D1-C03_Llama13B_zero2_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L13B-D1-C03_Llama13B_zero2_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L13B-D1-C04_Llama13B_zero3_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L13B-D1-C04_Llama13B_zero3_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L30B-D1-C05_Llama30B_zero2_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L30B-D1-C05_Llama30B_zero2_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L30B-D1-C06_Llama30B_zero3_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L30B-D1-C06_Llama30B_zero3_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L65B-D1-C07_Llama65B_zero2_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L65B-D1-C07_Llama65B_zero2_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L65B-D1-C07_Llama65B_zero3_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L65B-D1-C07_Llama65B_zero3_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L65B-M1-C05_Llama65B_megatron_tp2_pp8_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L65B-M1-C05_Llama65B_megatron_tp2_pp8_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L7B-D1-C01_Llama7B_zero2_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L7B-D1-C01_Llama7B_zero2_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L7B-D1-C02_Llama7B_zero3_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L7B-D1-C02_Llama7B_zero3_mbs1.csv -------------------------------------------------------------------------------- /workload/physical/model_workload/L7B-M1-C04_Llama7B_megatron_tp2_pp1_mbs1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/physical/model_workload/L7B-M1-C04_Llama7B_megatron_tp2_pp1_mbs1.csv -------------------------------------------------------------------------------- /workload/simAI/micro_test/all_gather.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/micro_test/all_gather.txt -------------------------------------------------------------------------------- /workload/simAI/micro_test/all_reduce.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/micro_test/all_reduce.txt -------------------------------------------------------------------------------- /workload/simAI/micro_test/all_to_all.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/micro_test/all_to_all.txt -------------------------------------------------------------------------------- /workload/simAI/micro_test/muti_all_reduce.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/micro_test/muti_all_reduce.txt -------------------------------------------------------------------------------- /workload/simAI/model_workload/G13B-M1-C01_GPT13B_megatron_tp8_pp1_mbs1_A100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/model_workload/G13B-M1-C01_GPT13B_megatron_tp8_pp1_mbs1_A100.txt -------------------------------------------------------------------------------- /workload/simAI/model_workload/G13B-M1-C02_GPT13B_megatron_tp8_pp1_mbs1_sp_A100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/model_workload/G13B-M1-C02_GPT13B_megatron_tp8_pp1_mbs1_sp_A100.txt -------------------------------------------------------------------------------- /workload/simAI/model_workload/G175B-M1-C03_GPT175B_megatron_tp8_pp1_mbs1_A100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/model_workload/G175B-M1-C03_GPT175B_megatron_tp8_pp1_mbs1_A100.txt -------------------------------------------------------------------------------- /workload/simAI/model_workload/L65B-M1-C05_Llama65B_megatron_tp8_pp1_mbs1_A100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/model_workload/L65B-M1-C05_Llama65B_megatron_tp8_pp1_mbs1_A100.txt -------------------------------------------------------------------------------- /workload/simAI/model_workload/L65B_D1_C08_Llama65B_deepspeed_zero3_A100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/model_workload/L65B_D1_C08_Llama65B_deepspeed_zero3_A100.txt -------------------------------------------------------------------------------- /workload/simAI/model_workload/L7B-D1-C02_Llama7B_deepspeed_zero3_A100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/model_workload/L7B-D1-C02_Llama7B_deepspeed_zero3_A100.txt -------------------------------------------------------------------------------- /workload/simAI/model_workload/L7B-M1-C04_Llama7B_megatron_tp2_pp1_mbs1_A100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload/simAI/model_workload/L7B-M1-C04_Llama7B_megatron_tp2_pp1_mbs1_A100.txt -------------------------------------------------------------------------------- /workload_applyer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_applyer.py -------------------------------------------------------------------------------- /workload_generator/SimAI_inference_workload_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/SimAI_inference_workload_generator.py -------------------------------------------------------------------------------- /workload_generator/SimAI_training_workload_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/SimAI_training_workload_generator.py -------------------------------------------------------------------------------- /workload_generator/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /workload_generator/analysis_pytorch_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/analysis_pytorch_trace.py -------------------------------------------------------------------------------- /workload_generator/generate_collective_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/generate_collective_test.py -------------------------------------------------------------------------------- /workload_generator/generate_deepspeed_stage1_2_workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/generate_deepspeed_stage1_2_workload.py -------------------------------------------------------------------------------- /workload_generator/generate_deepspeed_stage3_workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/generate_deepspeed_stage3_workload.py -------------------------------------------------------------------------------- /workload_generator/generate_ds_trace_replay_workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/generate_ds_trace_replay_workload.py -------------------------------------------------------------------------------- /workload_generator/generate_megatron_workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/generate_megatron_workload.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/MockedModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/MockedModel.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /workload_generator/mocked_model/inference/AiobDeepSeek.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/inference/AiobDeepSeek.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/inference/AiobQwen3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/inference/AiobQwen3.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/inference/MockedDeepSeek.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/inference/MockedDeepSeek.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/inference/MockedQwen3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/inference/MockedQwen3.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /workload_generator/mocked_model/training/AiobDeepSeek.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/training/AiobDeepSeek.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/training/AiobMegatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/training/AiobMegatron.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/training/MockedDeepSeek.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/training/MockedDeepSeek.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/training/MockedDeepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/training/MockedDeepspeed.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/training/MockedMegatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/mocked_model/training/MockedMegatron.py -------------------------------------------------------------------------------- /workload_generator/mocked_model/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /workload_generator/workload_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aliyun/aicb/HEAD/workload_generator/workload_generator.py --------------------------------------------------------------------------------