Skip to content

Commit

Permalink
Break up cutlass_extensions.cu, pt1 (#2944)
Browse files Browse the repository at this point in the history
Summary:
X-link: facebookresearch/FBGEMM#47

Pull Request resolved: #2944

- Break up cutlass_extensions.cu

Before - https://www.internalfb.com/buck2/86b77fae-04e1-44ca-b13f-d946417d6135: 8:14.5s

Differential Revision: D60874248
  • Loading branch information
q10 authored and facebook-github-bot committed Aug 7, 2024
1 parent 0ebb3ae commit 63e6f60
Show file tree
Hide file tree
Showing 6 changed files with 858 additions and 694 deletions.
3 changes: 3 additions & 0 deletions fbgemm_gpu/experimental/gen_ai/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ set(fbgemm_sources_include_directories
${CMAKE_CURRENT_SOURCE_DIR}/../..
${CMAKE_CURRENT_SOURCE_DIR}/../../include
${CMAKE_CURRENT_SOURCE_DIR}/../../../include
${CMAKE_CURRENT_SOURCE_DIR}/src/quantize
# PyTorch
${TORCH_INCLUDE_DIRS}
# Third-party
Expand All @@ -31,6 +32,8 @@ set(attention_ops_sources

set(quantize_ops_sources
src/quantize/cutlass_extensions.cu
src/quantize/cutlass_extensions/i8i8bf16.cu
src/quantize/cutlass_extensions/i8i8bf16_dynamic.cu
src/quantize/quantize.cu
src/quantize/quantize.cpp)

Expand Down
Loading

0 comments on commit 63e6f60

Please sign in to comment.