X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=speed.py;h=075b07ee3bb54deda3dc34b02cc05074d4d39381;hp=e5b0e3a6d13c777afbe6a58bc9cacd46e35293ff;hb=5ee7d04e8a0ba8fcf42f61c63ffaaad1b25b89d2;hpb=20285925e51c7adc6e7bb64bb9d2a5cab92c6aac

diff --git a/speed.py b/speed.py
index e5b0e3a..075b07e 100755
--- a/speed.py
+++ b/speed.py
@@ -42,4 +42,4 @@ for t in [ torch.float32, torch.float16 ]:
 
     except:
 
-        print(f'Cannot try with {t}')
+        print(f'{t} is not available on {device}')