Skip to content

Commit aaff90f

Browse files
datumboxfacebook-github-bot
authored andcommitted
[fbsync] remove dependency for dtype getters (#4291)
Reviewed By: fmassa Differential Revision: D30525890 fbshipit-source-id: f1056578d737c10b8d93d110952599000a773d1b
1 parent 69cf6b7 commit aaff90f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/common_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,11 +102,11 @@ def cycle_over(objs):
102102

103103

104104
def int_dtypes():
105-
return torch.testing.integral_types()
105+
return (torch.uint8, torch.int8, torch.int16, torch.int32, torch.int64)
106106

107107

108108
def float_dtypes():
109-
return torch.testing.floating_types()
109+
return (torch.float32, torch.float64)
110110

111111

112112
@contextlib.contextmanager

0 commit comments

Comments
 (0)