Skip to content

Commit

Permalink
add missing new files to CMakeLists.txt (pytorch#838)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#838

This was missed in D33358916 (pytorch@38a6c35)

Reviewed By: colin2328

Differential Revision: D33370387

fbshipit-source-id: 72007f51afd6757690a1898098e8b6207c3c487b
  • Loading branch information
jspark1105 authored and facebook-github-bot committed Dec 30, 2021
1 parent 0456e06 commit a508e46
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion fbgemm_gpu/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -222,8 +222,9 @@ set_source_files_properties(

set(fbgemm_gpu_sources_gpu
codegen/embedding_bounds_check.cu src/cumem_utils.cu
src/histogram_binning_calibration_ops.c, src/jagged_tensor_ops.cu
src/layout_transform_ops.cu src/permute_pooled_embedding_ops.cu
src/sparse_ops.cu src/split_embeddings_cache_cuda.cu)
src/quantize_ops.cu src/sparse_ops.cu src/split_embeddings_cache_cuda.cu)

set_source_files_properties(${fbgemm_gpu_sources_gpu}
PROPERTIES COMPILE_OPTIONS "${TORCH_CUDA_OPTIONS}")

0 comments on commit a508e46

Please sign in to comment.