flagos-ai · tianxiao-baai · Jul 28, 2025 · May 13, 2025 · Jun 4, 2025 · Jun 4, 2025
diff --git a/base/benchmarks/computation-INT8/iluvatar/BI150/README.md b/base/benchmarks/computation-INT8/iluvatar/BI150/README.md
@@ -0,0 +1,25 @@
+# 参评AI芯片信息
+
+* 厂商：ILUVATAR
+
+## 服务器1
+
+- 产品名称：BI150
+- 产品型号：BI150
+- TDP：350W
+
+# 所用服务器配置
+
+* 服务器数量：1
+
+## 服务器1
+
+* 单服务器内使用卡数：8
+* 服务器型号：R5300 G5
+* 操作系统版本：Ubuntu 20.04.6 LTS
+* 操作系统内核：linux5.4.0-148-generic
+* CPU：Intel(R) Xeon(R) Gold 6330 CPU @ 2.00GHz
+* docker版本：20.10.25
+* 内存：512GiB
+* 服务器间AI芯片直连规格及带宽：此评测样例无需服务器间通信
+
diff --git a/base/benchmarks/computation-INT8/iluvatar/BI150/case_config.yaml b/base/benchmarks/computation-INT8/iluvatar/BI150/case_config.yaml
@@ -0,0 +1,6 @@
+M: 5120
+N: 8192
+K: 8192
+DIST_BACKEND: "nccl"
+WARMUP: 1
+ITERS: 5
diff --git a/base/benchmarks/computation-INT8/iluvatar/BI150/env.sh b/base/benchmarks/computation-INT8/iluvatar/BI150/env.sh
@@ -0,0 +1,3 @@
+export PYTHONPATH=/usr/local/corex/lib64/python3/dist-packages:$PYTHONPATH
+export LD_LIBRARY_PATH=/usr/local/corex/lib64:$LD_LIBRARY_PATH
+export PATH=/usr/local/corex/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/corex/lib64/python3/dist-packages/bin:$PATH
diff --git a/base/benchmarks/computation-INT8/iluvatar/BI150/requirements.txt b/base/benchmarks/computation-INT8/iluvatar/BI150/requirements.txt
@@ -0,0 +1 @@
+#loguru
diff --git a/base/benchmarks/computation-INT8/main.py b/base/benchmarks/computation-INT8/main.py
@@ -70,9 +70,8 @@ def main(config, case_config, rank, world_size, local_rank):
     for _ in range(case_config.ITERS):
         _result = torch.mm(matrixA, matrixB)
 
-    if "iluvatar" not in config.vendor:
-        host_device_sync(config.vendor)
-        multi_device_sync(config.vendor)
+    host_device_sync(config.vendor)
+    multi_device_sync(config.vendor)
     end_time = time.perf_counter()
 
     exec_time = end_time - start_time
@@ -102,6 +101,8 @@ def main(config, case_config, rank, world_size, local_rank):
     for output_rank in range(config.node_size):
         if local_rank == output_rank:
             print(r"[FlagPerf Result]Rank {}'s computation-INT8=".format(dist.get_rank()) + str(result) + "TOPS")
+            if "iluvatar" in config.vendor:
+                print(r"[FlagPerf Result]Rank {} BI-V150 has 2 chips and overall GPU computation-INT8=".format(dist.get_rank()) + str(result*2) + "TOPS")
         multi_device_sync(config.vendor)
 
     dist.destroy_process_group()

diff --git a/base/vendors/iluvatar/pytorch_2.1/Dockerfile b/base/vendors/iluvatar/pytorch_2.1/Dockerfile
@@ -1,4 +1,5 @@
-FROM zibo.harbor.iluvatar.com.cn:30000/saas/mr-bi150-4.2.0-x86-ubuntu20.04-py3.10-customer-ds-671b:v1.3
+FROM flagperf_iluvatar:v1.0
+#FROM zibo.harbor.iluvatar.com.cn:30000/saas/mr-bi150-4.2.0-x86-ubuntu20.04-py3.10-customer-ds-671b:v1.3
 #RUN /bin/bash -c "pip config set global.index-url https://mirror.baidu.com/pypi/simple"
 #RUN /bin/bash -c "uname -a"
 #RUN /bin/bash -c alias python3=python