fix syntax error in bfloat16 PR (#64122)
authorRishi Puri <puririshi98@berkeley.edu>
Tue, 31 Aug 2021 20:47:29 +0000 (13:47 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 31 Aug 2021 21:33:12 +0000 (14:33 -0700)
commit13484084a64df5f2c5deea26b8cc2f30833038f3
tree0abf26aa3d55d47466d16de9b99104c3f15dfc69
parent8d08b103be936d78d5d4ed90c0547aeccb8ce166
fix syntax error in bfloat16 PR (#64122)

Summary:
fixes prior syntax error from PR ngimel

Pull Request resolved: https://github.com/pytorch/pytorch/pull/64122

Reviewed By: H-Huang

Differential Revision: D30643596

Pulled By: ngimel

fbshipit-source-id: 0a2d5a40fb6dc7339cd03112e57ef0e1bf8a000e
test/test_cuda.py
torch/cuda/__init__.py