[KUNLUNXIN] add FlagGems blas op support & minor fix for env vars (#727)

Co-authored-by: w4yne <[email protected]>
FlagOpen · Sep 11, 2024 · a843a8b · a843a8b
1 parent a76527b
commit a843a8b
Show file tree

Hide file tree

Showing 11 changed files with 43 additions and 3 deletions.
diff --git a/operation/benchmarks/addmm/kunlunxin/R300p/case_config.yaml b/operation/benchmarks/addmm/kunlunxin/R300p/case_config.yaml
@@ -0,0 +1,5 @@
+M : 16
+N : 40
+K : 256
+ITERS: 50
+SPECTFLOPS: 9999
diff --git a/operation/benchmarks/addmm/kunlunxin/R300p/env.sh b/operation/benchmarks/addmm/kunlunxin/R300p/env.sh
@@ -0,0 +1,5 @@
+echo "KUNLUNXIN ENV.SH start"
+
+source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda
+
+echo "KUNLUNXIN ENV.SH end"
diff --git a/operation/benchmarks/bmm/kunlunxin/R300p/case_config.yaml b/operation/benchmarks/bmm/kunlunxin/R300p/case_config.yaml
@@ -0,0 +1,5 @@
+M : 16
+N : 40
+K : 256
+ITERS: 50
+SPECTFLOPS: 9999
diff --git a/operation/benchmarks/bmm/kunlunxin/R300p/env.sh b/operation/benchmarks/bmm/kunlunxin/R300p/env.sh
@@ -0,0 +1,5 @@
+echo "KUNLUNXIN ENV.SH start"
+
+source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda
+
+echo "KUNLUNXIN ENV.SH end"
diff --git a/operation/benchmarks/cross_entropy_loss/kunlunxin/R300p/env.sh b/operation/benchmarks/cross_entropy_loss/kunlunxin/R300p/env.sh
@@ -1,6 +1,6 @@
 echo "KUNLUNXIN ENV.SH start"
 
 source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda
-export XPU_enable_reorder=1
+export TRITONXPU_BUFFER_SIZE=128
 
 echo "KUNLUNXIN ENV.SH end"
diff --git a/operation/benchmarks/linear/kunlunxin/R300p/case_config.yaml b/operation/benchmarks/linear/kunlunxin/R300p/case_config.yaml
@@ -0,0 +1,5 @@
+M : 16
+N : 40
+K : 256
+ITERS: 50
+SPECTFLOPS: 9999
diff --git a/operation/benchmarks/linear/kunlunxin/R300p/env.sh b/operation/benchmarks/linear/kunlunxin/R300p/env.sh
@@ -0,0 +1,5 @@
+echo "KUNLUNXIN ENV.SH start"
+
+source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda
+
+echo "KUNLUNXIN ENV.SH end"
diff --git a/operation/benchmarks/max/kunlunxin/R300p/env.sh b/operation/benchmarks/max/kunlunxin/R300p/env.sh
@@ -1,6 +1,6 @@
 echo "KUNLUNXIN ENV.SH start"
 
 source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda
-export Triton_big_instcombine=1000
+export INST_COMBINE_LOOP_THRESHOLD=1000
 
 echo "KUNLUNXIN ENV.SH end"
diff --git a/operation/benchmarks/min/kunlunxin/R300p/env.sh b/operation/benchmarks/min/kunlunxin/R300p/env.sh
@@ -1,6 +1,6 @@
 echo "KUNLUNXIN ENV.SH start"
 
 source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda
-export Triton_big_instcombine=1000
+export INST_COMBINE_LOOP_THRESHOLD=1000
 
 echo "KUNLUNXIN ENV.SH end"
diff --git a/operation/benchmarks/mm/kunlunxin/R300p/case_config.yaml b/operation/benchmarks/mm/kunlunxin/R300p/case_config.yaml
@@ -0,0 +1,5 @@
+M : 16
+N : 40
+K : 256
+ITERS: 50
+SPECTFLOPS: 9999
diff --git a/operation/benchmarks/mm/kunlunxin/R300p/env.sh b/operation/benchmarks/mm/kunlunxin/R300p/env.sh
@@ -0,0 +1,5 @@
+echo "KUNLUNXIN ENV.SH start"
+
+source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda
+
+echo "KUNLUNXIN ENV.SH end"