Re-organize TBE tests, pt 1 (pytorch#2263)

Summary:
Pull Request resolved: pytorch#2263

- Re-organize TBE tests, pt 1

Reviewed By: spcyppt

Differential Revision: D52714927

fbshipit-source-id: bd9b80c9db6b33e28770b00f58cae3af16e2b2b7

.github/scripts/fbgemm_gpu_test.bash

@@ -84,19 +84,19 @@ run_fbgemm_gpu_tests () {
 
   # These are either non-tests or currently-broken tests in both FBGEMM_GPU and FBGEMM_GPU-CPU
   local files_to_skip=(
-    ./test_utils.py
-    ./split_table_batched_embeddings_test.py
+    ./tbe/split_table_batched_embeddings_test.py
     ./ssd_split_table_batched_embeddings_test.py
   )
 
   if [ "$fbgemm_variant" == "cpu" ]; then
     # These are tests that are currently broken in FBGEMM_GPU-CPU
     local ignored_tests=(
+      ./tbe/forward_test.py
       ./uvm_test.py
     )
   elif [ "$fbgemm_variant" == "rocm" ]; then
-    # https://github.com/pytorch/FBGEMM/issues/1559
     local ignored_tests=(
+      # https://github.com/pytorch/FBGEMM/issues/1559
       ./batched_unary_embeddings_test.py
     )
   else

fbgemm_gpu/test/tbe/__init__.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python3
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.