aihwkit icon indicating copy to clipboard operation
aihwkit copied to clipboard
verified publisher icon IBM

Fix the support of different dtypes for the torch model

Open jubueche opened this issue 1 year ago • 1 comments

Related issues

Issue #623

Description

Changed the .to function mostly.

Details

  • Verification on GPU needed
  • A tests needs to be added that verifies the output correctness against pure torch.

jubueche avatar Mar 08 '24 17:03 jubueche

Added tests for the torch version and verified that it is working on a A100 GPU. Also, I added a line in the helper.py to make make mypy pass with a newer mypy version.

jubueche avatar Mar 11 '24 09:03 jubueche