updatekunlunxin transformer_xl config

FlagOpen · Nov 2, 2023 · c348a5f · c348a5f
1 parent 3e3a18d
commit c348a5f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/training/kunlunxin/transformer_xl-pytorch/config/environment_variables.sh b/training/kunlunxin/transformer_xl-pytorch/config/environment_variables.sh
@@ -1,2 +1,6 @@
-export XACC_ENABLE=1
+export XACC=1
+
 export BKCL_PCIE_RING=1
+export BKCL_TIMEOUT=1800
+
+export XMLIR_D_XPU_L3_SIZE=66060288
diff --git a/training/kunlunxin/transformer_xl-pytorch/extern/.gitkeep b/training/kunlunxin/transformer_xl-pytorch/extern/.gitkeep
diff --git a/training/run_benchmarks/config/test_conf.py b/training/run_benchmarks/config/test_conf.py
@@ -82,6 +82,7 @@
 
     # "bert_hf:pytorch_1.13:A100:1:8:1": "/raid/dataset/bert_hf_train",
     # "longformer:pytorch_1.12:A100:1:8:1": "/raid/dataset/longformer_train/",
+    # "transformer_xl:pytorch_1.13:A100:1:8:1": "/raid/dataset/transformer_xl/",
     # "detr:pytorch_1.13:A100:1:8:1": "/raid/dataset/detr/coco2017/",
 
     # "llama1_7B:paddle_2.5.1:TP1PP1SH2SP8A10040G:1:8:1":"/raid/dataset/llama/"
@@ -105,5 +106,6 @@
     # kunlunxin cases
     # "gpt2:pytorch:R300:1:8:1": "/raid/dataset/gpt2"
     # "resnet50:pytorch:R300:1:8:1": "/raid/dataset/ImageNet_1k_2012/",
+    # "transformer_xl:pytorch:R300:1:8:1": "/raid/dataset/transformer_xl/",
 }