Skip to content

Conversation

@wyli
Copy link
Contributor

@wyli wyli commented Oct 15, 2021

Signed-off-by: Wenqi Li wenqil@nvidia.com

Description

Fixes issue from nightly test:

======================================================================
ERROR: test_train_timing (__main__.IntegrationFastTrain)
----------------------------------------------------------------------
AttributeError: module 'torch.cuda.amp' has no attribute 'autocast'

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/home/jenkins/agent/workspace/Monai-pytorch-versions/tests/utils.py", line 499, in _wrapper
    raise RuntimeError(res.traceback) from res
RuntimeError: Traceback (most recent call last):
  File "/home/jenkins/agent/workspace/Monai-pytorch-versions/tests/utils.py", line 451, in run_process
    output = func(*args, **kwargs)
  File "/home/jenkins/agent/workspace/Monai-pytorch-versions/tests/utils.py", line 524, in _call_original_func
    return f(*args, **kwargs)
  File "/home/jenkins/agent/workspace/Monai-pytorch-versions/tests/test_integration_fast_train.py", line 191, in test_train_timing
    with torch.cuda.amp.autocast():
AttributeError: module 'torch.cuda.amp' has no attribute 'autocast'


----------------------------------------------------------------------

Status

Ready

Types of changes

  • Non-breaking change (fix or new feature that would not break existing functionality).
  • Breaking change (fix or new feature that would cause existing functionality to change).
  • New tests added to cover the changes.
  • Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
  • Quick tests passed locally by running ./runtests.sh --quick --unittests.
  • In-line docstrings updated.
  • Documentation updated, tested make html command in the docs/ folder.

Signed-off-by: Wenqi Li <wenqil@nvidia.com>
@wyli wyli requested a review from Nic-Ma October 15, 2021 08:36
@wyli wyli changed the title skip integration tests for torch 1.5 skip tests/test_integration_fast_train.py for torch 1.5 Oct 15, 2021
@wyli wyli requested review from ericspod and rijobro October 15, 2021 08:36
@wyli
Copy link
Contributor Author

wyli commented Oct 15, 2021

/build

@wyli wyli enabled auto-merge (squash) October 15, 2021 09:24
@wyli wyli merged commit ff08334 into Project-MONAI:dev Oct 15, 2021
@wyli wyli deleted the fixes-tests branch October 15, 2021 10:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants