Skip to content

Commit

Permalink
Integrate gen_ai ops into FBGEMM_GPU OSS build (#2512)
Browse files Browse the repository at this point in the history
Summary:
- Integrate gen_ai ops and tests into FBGEMM_GPU OSS build
- Separate out FBGEMM_GPU main module-specific build instructions into separate file

Pull Request resolved: #2512

Reviewed By: spcyppt

Differential Revision: D56327795

Pulled By: q10
  • Loading branch information
q10 authored and facebook-github-bot committed Apr 19, 2024
1 parent dbae12b commit 7f5b721
Show file tree
Hide file tree
Showing 9 changed files with 709 additions and 612 deletions.
6 changes: 6 additions & 0 deletions .github/scripts/fbgemm_gpu_test.bash
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,12 @@ test_all_fbgemm_gpu_modules () {
fbgemm_gpu/experimental/example/test
)

if [ "$fbgemm_variant" == "cuda" ]; then
target_directories+=(
fbgemm_gpu/experimental/gen_ai/test
)
fi

for test_dir in "${target_directories[@]}"; do
cd "${test_dir}" || return 1
run_fbgemm_gpu_tests "${env_name}" "${fbgemm_variant}" || return 1
Expand Down
Loading

0 comments on commit 7f5b721

Please sign in to comment.