We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5929e7c commit addc0e3Copy full SHA for addc0e3
advanced_source/semi_structured_sparse.py
@@ -210,6 +210,8 @@
210
SparseSemiStructuredTensor._FORCE_CUTLASS = True
211
torch.manual_seed(100)
212
213
+# Set default device to "cuda:0"
214
+torch.set_default_device(torch.device("cuda:0" if torch.cuda.is_available() else "cpu"))
215
216
######################################################################
217
# We’ll also need to define some helper functions that are specific to the
0 commit comments