From b130e1b2f6d8fe3e504e4a1b252839bcd1110553 Mon Sep 17 00:00:00 2001 From: Codersheepchen Date: Wed, 3 Sep 2025 04:04:17 -0400 Subject: [PATCH] merge embedding ut --- tensorflow/core/kernels/BUILD | 134 +++------------------------------- 1 file changed, 11 insertions(+), 123 deletions(-) diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 014c92711..f74da7e88 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -4094,25 +4094,6 @@ tf_kernel_library( deps = MATH_DEPS, ) -tf_cc_test( - name = "embedding_fused_gather_test", - srcs = if_enable_annc([ - "embedding_fused_gather_test.cc", - ]), - deps = [ - ":ops_testutil", - ":ops_util", - ":embedding_fused_gather_op", - "//tensorflow/core:core_cpu", - "//tensorflow/core:framework", - "//tensorflow/core:lib", - "//tensorflow/core:protos_all_cc", - "//tensorflow/core:test", - "//tensorflow/core:test_main", - "//tensorflow/core:testlib", - ], -) - tf_kernel_library( name = "embedding_fused_padding_op", srcs = if_enable_annc([ @@ -4129,26 +4110,6 @@ tf_kernel_library( deps = MATH_DEPS, ) -tf_cc_test( - name = "embedding_fused_sparse_dynamic_stitch_test", - srcs = if_enable_annc([ - "embedding_fused_sparse_dynamic_stitch_test.cc", - ]), - deps = [ - ":embedding_fused_sparse_dynamic_stitch_op", - ":host_constant_op", - ":ops_testutil", - ":ops_util", - "//tensorflow/core:core_cpu", - "//tensorflow/core:framework", - "//tensorflow/core:lib", - "//tensorflow/core:protos_all_cc", - "//tensorflow/core:test", - "//tensorflow/core:test_main", - "//tensorflow/core:testlib", - ], -) - tf_kernel_library( name = "embedding_fused_reshape_op", srcs = if_enable_annc([ @@ -4159,25 +4120,6 @@ tf_kernel_library( ], ) -tf_cc_test( - name = "embedding_fused_sparse_reshape_test", - srcs = if_enable_annc([ - "embedding_fused_sparse_reshape_test.cc", - ]), - deps = [ - ":ops_testutil", - ":ops_util", - ":embedding_fused_reshape_op", - "//tensorflow/core:core_cpu", - "//tensorflow/core:framework", - "//tensorflow/core:lib", - "//tensorflow/core:protos_all_cc", - "//tensorflow/core:test", - "//tensorflow/core:test_main", - "//tensorflow/core:testlib", - ], -) - tf_kernel_library( name = "embedding_fused_sparse_segment_reduce_op", srcs = if_enable_annc([ @@ -4186,26 +4128,6 @@ tf_kernel_library( deps = MATH_DEPS, ) -tf_cc_test( - name = "embedding_fused_sparse_segment_reduce_test", - srcs = if_enable_annc([ - "embedding_fused_sparse_segment_reduce_test.cc", - ]), - deps = [ - ":embedding_fused_sparse_segment_reduce_op", - ":host_constant_op", - ":ops_testutil", - ":ops_util", - "//tensorflow/core:core_cpu", - "//tensorflow/core:framework", - "//tensorflow/core:lib", - "//tensorflow/core:protos_all_cc", - "//tensorflow/core:test", - "//tensorflow/core:test_main", - "//tensorflow/core:testlib", - ], -) - tf_kernel_library( name = "embedding_fused_sparse_segment_reduce_nonzero_op", srcs = if_enable_annc([ @@ -4214,26 +4136,6 @@ tf_kernel_library( deps = MATH_DEPS, ) -tf_cc_test( - name = "embedding_fused_sparse_segment_reduce_nonzero_test", - srcs = if_enable_annc([ - "embedding_fused_sparse_segment_reduce_nonzero_test.cc", - ]), - deps = [ - ":embedding_fused_sparse_segment_reduce_nonzero_op", - ":host_constant_op", - ":ops_testutil", - ":ops_util", - "//tensorflow/core:core_cpu", - "//tensorflow/core:framework", - "//tensorflow/core:lib", - "//tensorflow/core:protos_all_cc", - "//tensorflow/core:test", - "//tensorflow/core:test_main", - "//tensorflow/core:testlib", - ], -) - tf_kernel_library( name = "embedding_fused_sparse_select_op", srcs = if_enable_annc([ @@ -4242,25 +4144,6 @@ tf_kernel_library( deps = MATH_DEPS, ) -tf_cc_test( - name = "embedding_fused_sparse_select_test", - srcs = if_enable_annc([ - "embedding_fused_sparse_select_test.cc", - ]), - deps = [ - ":ops_testutil", - ":ops_util", - ":embedding_fused_sparse_select_op", - "//tensorflow/core:core_cpu", - "//tensorflow/core:framework", - "//tensorflow/core:lib", - "//tensorflow/core:protos_all_cc", - "//tensorflow/core:test", - "//tensorflow/core:test_main", - "//tensorflow/core:testlib", - ], -) - cc_library( name = "embedding_fused_ops", deps = if_enable_annc([ @@ -4275,17 +4158,22 @@ cc_library( ]), ) -tf_cuda_cc_test( - name = "embedding_fused_padding_test", +tf_cc_test( + name = "embedding_fused_ops_test", size = "small", srcs = if_enable_annc([ + "embedding_fused_sparse_dynamic_stitch_test.cc", + "embedding_fused_sparse_segment_reduce_test.cc", + "embedding_fused_sparse_segment_reduce_nonzero_test.cc", "embedding_fused_padding_test.cc", + "embedding_fused_sparse_select_test.cc", + "embedding_fused_gather_test.cc", + "embedding_fused_sparse_reshape_test.cc", ]), - deps = if_enable_annc([ - ":embedding_fused_padding_op", - ":host_constant_op", + deps = [ ":ops_testutil", ":ops_util", + ":embedding_fused_ops", "//tensorflow/core:core_cpu", "//tensorflow/core:framework", "//tensorflow/core:lib", @@ -4293,7 +4181,7 @@ tf_cuda_cc_test( "//tensorflow/core:test", "//tensorflow/core:test_main", "//tensorflow/core:testlib", - ]), + ], ) tf_kernel_library( -- Gitee