Skip to content

Commit

Permalink
Re-sync with internal repository (pytorch#468)
Browse files Browse the repository at this point in the history
Co-authored-by: Facebook Community Bot <[email protected]>
  • Loading branch information
facebook-github-bot and facebook-github-bot authored Jan 5, 2021
1 parent dd35aba commit f525afd
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions fbgemm_gpu/defs.bzl
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

def get_fbgemm_gpu_wrapper_srcs():
return [
"src/embedding_wrappers.cu",
"src/quantize_wrappers.cu",
"src/sparse_wrappers.cu",
]

def get_fbgemm_gpu_public_headers():
return [
"include/fbgemm_gpu/batched_unary_embeddings.cuh",
"include/fbgemm_gpu/bench_utils.cuh",
"include/fbgemm_gpu/cuda_utils.cuh",
"include/fbgemm_gpu/embedding_wrappers.cuh",
"include/fbgemm_gpu/quantize_ops.cuh",
"include/fbgemm_gpu/quantize_wrappers.cuh",
"include/fbgemm_gpu/sparse_ops.cuh",
"include/fbgemm_gpu/sparse_wrappers.cuh",
]

0 comments on commit f525afd

Please sign in to comment.