Skip to content

Commit

Permalink
fix for OSS CMakefile build for merge pooled embedding (#1444)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #1444

Currently merge pooled embedding op is not covered by OSS tests due to nvml dependency.

Reviewed By: jspark1105

Differential Revision: D40952247

fbshipit-source-id: fdae4b2d192f0097070bf963d79d5a84e57e24b0
  • Loading branch information
jianyuh authored and facebook-github-bot committed Nov 2, 2022
1 parent 0fc01bb commit 7935a81
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions fbgemm_gpu/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -314,8 +314,12 @@ if(NOT FBGEMM_CPU_ONLY)
src/metric_ops_host.cpp)

if(NVML_LIB_PATH)
list(APPEND fbgemm_gpu_sources_cpu src/merge_pooled_embeddings_cpu.cpp
src/merge_pooled_embeddings_gpu.cpp)
list(
APPEND
fbgemm_gpu_sources_cpu
src/merge_pooled_embeddings_cpu.cpp
src/merge_pooled_embeddings_gpu.cpp
src/topology_utils.cpp)
endif()
endif()

Expand Down

0 comments on commit 7935a81

Please sign in to comment.