Skip to content

Commit

Permalink
Update sparse_ops.py to use generic gpu target fbgemm_gpu:input_combi…
Browse files Browse the repository at this point in the history
…ne 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
  • Loading branch information
joebos authored and facebook-github-bot committed Jul 26, 2024
1 parent 15a4de5 commit a300568
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions fbgemm_gpu/fbgemm_gpu/sparse_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,6 @@
torch.ops.load_library(
"//deeplearning/fbgemm/fbgemm_gpu/codegen:embedding_ops_hip"
)
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine_hip")
torch.ops.load_library(
"//deeplearning/fbgemm/fbgemm_gpu/codegen:index_select_ops_hip"
)
Expand All @@ -36,7 +35,8 @@
"//deeplearning/fbgemm/fbgemm_gpu:merge_pooled_embeddings"
)
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu/codegen:embedding_ops")
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine")

torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:input_combine")

torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:sparse_ops_cpu")
torch.ops.load_library(
Expand Down

0 comments on commit a300568

Please sign in to comment.