Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Upgrade to HPU release 1.7.0 (#15616) 

Signed-off-by: Jerome <janand@habana.ai>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
This commit is contained in:
Jerome Anand 2022-11-10 15:17:17 +05:30 committed by GitHub
parent 2076746238
commit e79a69a9ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.azure/hpu-tests.yml
View File

@ -35,7 +35,7 @@ jobs:
cancelTimeoutInMinutes: "2"
pool: intel-hpus
container:
image: "vault.habana.ai/gaudi-docker/1.6.1/ubuntu20.04/habanalabs/pytorch-installer-1.12.0:latest"
image: "vault.habana.ai/gaudi-docker/1.7.0/ubuntu20.04/habanalabs/pytorch-installer-1.12.0:latest"
options: "--runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --ipc=host --shm-size=4g -v /usr/bin/docker:/tmp/docker:ro"
workspace:
clean: all

2
dockers/ci-runner-hpu/Dockerfile
View File

@ -16,7 +16,7 @@
# gaudi-docker-agent:latest
ARG DIST="latest"
ARG GAUDI_VERSION="1.6.1"
ARG GAUDI_VERSION="1.7.0"
ARG PYTORCH_INSTALLER_VERSION="1.12.0"
FROM vault.habana.ai/gaudi-docker/${GAUDI_VERSION}/ubuntu20.04/habanalabs/pytorch-installer-${PYTORCH_INSTALLER_VERSION}:${DIST}

2
tests/tests_pytorch/plugins/precision/hpu/test_hpu.py
View File

@ -85,7 +85,7 @@ def test_pure_half_precision(tmpdir, hmp_params: dict):
assert isinstance(trainer.strategy.precision_plugin, HPUPrecisionPlugin)
assert trainer.strategy.precision_plugin.precision == 16
with pytest.raises(SystemExit):
with pytest.raises(RuntimeError, match=r"float16/half is not supported on Gaudi."):
trainer.fit(model)