Searched refs:multi_processor_count (Results 1 – 10 of 10) sorted by relevance
106 multi_processor_count: Optional[int] = None variable in DeviceProperties124 multi_processor_count=props.multi_processor_count,
275 assert device_prop.multi_processor_count326 <= max_blocks_per_sm * device_prop.multi_processor_count
12 self.multi_processor_count = 80
45 if warn_imbalance(lambda props: props.multi_processor_count):
73 grid_size = min(grid_size, device_property.multi_processor_count * blocks_per_sm)
1468 "parallelism": capability.multi_processor_count,
1046 avail_sms = torch.cuda.get_device_properties(index).multi_processor_count
710 num_sm = device_properties.multi_processor_count
1908 multi_processor_count: _int
941 return x + prop.multi_processor_count