Skip to content

Commit a300568

Browse files
joebosfacebook-github-bot
authored andcommitted
Update sparse_ops.py to use generic gpu target fbgemm_gpu:input_combine to support both nvidia and AMD (#2905)
Summary: Pull Request resolved: #2905 Update sparse_ops.py to use generic gpu target fbgemm_gpu:input_combine to support both nvidia and AMD Reviewed By: Yuzhen11 Differential Revision: D60271751 fbshipit-source-id: 17aeb321ebf805589c6e8f3f3698220725521a2e
1 parent 15a4de5 commit a300568

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fbgemm_gpu/fbgemm_gpu/sparse_ops.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@
2626
torch.ops.load_library(
2727
"//deeplearning/fbgemm/fbgemm_gpu/codegen:embedding_ops_hip"
2828
)
29-
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine_hip")
3029
torch.ops.load_library(
3130
"//deeplearning/fbgemm/fbgemm_gpu/codegen:index_select_ops_hip"
3231
)
@@ -36,7 +35,8 @@
3635
"//deeplearning/fbgemm/fbgemm_gpu:merge_pooled_embeddings"
3736
)
3837
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu/codegen:embedding_ops")
39-
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine")
38+
39+
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine")
4040

4141
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:sparse_ops_cpu")
4242
torch.ops.load_library(

0 commit comments

Comments
 (0)