Skip to content

Commit

Permalink
remove apex
Browse files Browse the repository at this point in the history
  • Loading branch information
mvpatel2000 committed Feb 26, 2024
1 parent ad2191e commit 2d6d73e
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 24 deletions.
20 changes: 0 additions & 20 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -270,26 +270,6 @@ RUN if [ -n "$MOFED_VERSION" ] ; then \
rm -rf /tmp/mofed ; \
fi


#####################
# Install NVIDIA Apex
#####################
# skip if torch nightly is installed as there is incompatability
RUN if [[ -n "$CUDA_VERSION" ]] && [[ -z "${PYTORCH_NIGHTLY_URL}" ]]; then \
mkdir -p /tmp/apex && \
cd /tmp/apex && \
git clone https://github.com/NVIDIA/apex && \
cd apex && \
git checkout 82ee367f3da74b4cd62a1fb47aa9806f0f47b58b && \
pip${PYTHON_VERSION} install --no-cache-dir -r requirements.txt && \
pip${PYTHON_VERSION} install --no-cache-dir \
--global-option="--cpp_ext" \
--global-option="--cuda_ext" \
--target /usr/local/lib/python${PYTHON_VERSION}/dist-packages \
./ && \
rm -rf /tmp/apex ; \
fi

##########################
# Install Flash Attention
##########################
Expand Down
4 changes: 0 additions & 4 deletions tests/test_passes.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,10 +84,6 @@ class TestAlgorithmOrderingPasses:
@pytest.mark.parametrize('algorithm_cls', [LowPrecisionLayerNorm])
def test_algorithm_last(self, algorithm_cls: Type[Algorithm], always_match_algorithms: List[Algorithm],
dummy_logger: Logger, dummy_state: State):

if algorithm_cls == LowPrecisionLayerNorm:
pytest.importorskip('apex')

algorithm = algorithm_cls()
algorithm.apply = Mock(return_value='algo')
algorithm.match = Mock(return_value=True)
Expand Down

0 comments on commit 2d6d73e

Please sign in to comment.