Searched defs:SM90OrLater (Results 1 – 2 of 2) sorted by relevance
32 SM90OrLater = LazyVal(lambda: torch.cuda.is_available() and torch.cuda.get_device_capability() >= (… variable
38 SM90OrLater = SM90OrLater and not torch.version.hip variable