Skip to content

Update for PyTorch 1.8
Compare
Choose a tag to compare
@mmuckley mmuckley released this 09 Apr 23:22
v1.2.0
4d98054
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

This updates torchkbnufft for PyTorch version 1.8. It uses a new version of index_add that operates natively on complex tensors. The update also fixes a performance regression that arose due to thread management as identified in Issue #25.

Most changes came from PR #27, which has the below list of modifications:

  • Update requirements.txt and dev-requirements.txt to latest packages.
  • Remove calc_split_sizes - we can now use tensor_split.
  • Removed some calls to tensor attributes - these can be expensive.
  • Removal of kwarg usage for some torch.jit.script functions - these can behave strangely with scripted functions.
  • Removal of index_put for accumulation. We now only use index_add.
Assets 2
Loading