Redefine FBGEMM targets with gpu_cpp_library [10/N] (#3131)

Summary:
Pull Request resolved: #3131

X-link: facebookresearch/FBGEMM#218

- Redefine `input_combine_*` targets using `gpu_cpp_library`

Reviewed By: spcyppt

Differential Revision: D62611707

fbshipit-source-id: f6ddbb989eaa99b82925e25e1522b8a4718b396b

fbgemm_gpu/src/placeholder.cpp

@@ -6,7 +6,6 @@
  * LICENSE file in the root directory of this source tree.
  */
 
-/*
-    This is placeholder code to force compilation and generation of .so file.
-*/
+/// This is a placeholder source file that is used to force compilation and
+/// generation of an .SO file.
 namespace fbgemm_gpu {}

fbgemm_gpu/test/combine/common.py

@@ -11,14 +11,16 @@
 import fbgemm_gpu
 import torch
 
+from fbgemm_gpu.utils.loader import load_torch_module
+
 # pyre-fixme[16]: Module `fbgemm_gpu` has no attribute `open_source`.
 open_source: bool = getattr(fbgemm_gpu, "open_source", False)
 
 if not open_source:
-    if torch.version.hip:
-        torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine_hip")
-    else:
-        torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine")
+    load_torch_module(
+        "//deeplearning/fbgemm/fbgemm_gpu:input_combine",
+        hip_path="//deeplearning/fbgemm/fbgemm_gpu:input_combine_hip",
+    )
     torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine_cpu")