From 2a3a8e81525ccb086cb586a94f29677eac00019e Mon Sep 17 00:00:00 2001
From: wenxindongwork <161090399+wenxindongwork@users.noreply.github.com>
Date: Mon, 25 Nov 2024 10:58:24 -0800
Subject: [PATCH] Use smaller embedding size as 32 leads to segmentation fault
 (#487)

---
 dags/solutions_team/solutionsteam_tf_nightly_supported.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dags/solutions_team/solutionsteam_tf_nightly_supported.py b/dags/solutions_team/solutionsteam_tf_nightly_supported.py
index 13abe496..c2bebeed 100644
--- a/dags/solutions_team/solutionsteam_tf_nightly_supported.py
+++ b/dags/solutions_team/solutionsteam_tf_nightly_supported.py
@@ -112,7 +112,7 @@
       runtime_version=RuntimeVersion.TPU_VM_TF_NIGHTLY_POD.value,
   )
 
-  embedding_dim = 32
+  embedding_dim = 16
   tf_dlrm_v5p_8 = tf_config.get_tf_dlrm_config(
       project_name=Project.TPU_PROD_ENV_AUTOMATED.value,
       tpu_version=TpuVersion.V5P,