add test case

alibaba · Dec 28, 2023 · 0557c9b · 0557c9b
1 parent e524243
commit 0557c9b
Show file tree

Hide file tree

Showing 4 changed files with 587 additions and 0 deletions.
diff --git a/docs/source/component/backbone.md b/docs/source/component/backbone.md
@@ -1091,6 +1091,7 @@ MovieLens-1M数据集效果：
 
 - Highway Network: [highway network](../models/highway.md)
 - Cross Decoupling Network: [CDN](../models/cdn.md)
+- DLRM+SENet: [dlrm_senet_on_criteo.config](https://github.com/alibaba/EasyRec/tree/master/examples/configs/dlrm_senet_on_criteo.config)
 
 # 组件库介绍
 

diff --git a/easy_rec/python/layers/keras/layer_norm.py b/easy_rec/python/layers/keras/layer_norm.py
@@ -211,6 +211,8 @@ def _fused_can_be_used(self, ndims):
     Check if the axis is contiguous and can be collapsed into the last axis.
     The self.axis is assumed to have no duplicates.
     """
+    if not tf.test.is_gpu_available():
+      return False
     axis = sorted(self.axis)
     can_use_fused = False