Skip to content

Commit

Permalink
[fbgemm_gpu] Fix mpmath dependency
Browse files Browse the repository at this point in the history
- Fix mpmath dependency to allow tests to pass again; see

jianyicheng/mase-docker#9
nod-ai/SHARK-Studio#2095
  • Loading branch information
q10 committed Feb 26, 2024
1 parent 2bba5cc commit a4f874c
Show file tree
Hide file tree
Showing 4 changed files with 29 additions and 2 deletions.
6 changes: 4 additions & 2 deletions .github/scripts/utils_build.bash
Original file line number Diff line number Diff line change
Expand Up @@ -247,8 +247,10 @@ install_build_tools () {

echo "[INSTALL] Installing build tools ..."
# NOTES:
# - Only the openblas package will install cblas.h directly into
# $CONDA_PREFIX/include directory
#
# - Only the openblas package will install <cblas.h> directly into
# $CONDA_PREFIX/include directory, which is required for FBGEMM tests
#
# - ncurses is needed to silence bad libtinfo6.so errors for ROCm+Clang builds
#
# shellcheck disable=SC2086
Expand Down
6 changes: 6 additions & 0 deletions docs/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,8 @@
# Copyright (c) Meta Platforms, Inc. and 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.

breathe
sphinx_rtd_theme
6 changes: 6 additions & 0 deletions fbgemm_gpu/docs/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
# Copyright (c) Meta Platforms, Inc. and 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.

# Note: Sphinx 7+ currently runs into an `Undefinederror("'style' is undefined")`
# We should eventually move to Sphinx 7+ to resolve
# https://github.com/sphinx-doc/sphinx/issues/1514
Expand Down
13 changes: 13 additions & 0 deletions fbgemm_gpu/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,19 @@
# Copyright (c) Meta Platforms, Inc. and 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.

# NOTES:
#
# - A fixed version of mpmath is needed to work around an AttributeError; see:
# * https://github.com/nod-ai/SHARK/issues/2095
# * https://github.com/jianyicheng/mase-docker/pull/9

cmake
hypothesis
jinja2
mpmath==1.3.0
ninja
numpy
scikit-build
Expand Down

0 comments on commit a4f874c

Please sign in to comment.