Home
last modified time | relevance | path

Searched refs:max_threads_per_multi_processor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/runtime/
H A Dhints.py105 max_threads_per_multi_processor: Optional[int] = None variable in DeviceProperties
123 max_threads_per_multi_processor=props.max_threads_per_multi_processor,
H A Dtriton_heuristics.py274 assert device_prop.max_threads_per_multi_processor
304 // device_prop.max_threads_per_multi_processor
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/triton/
H A Ddevice_interface.py11 self.max_threads_per_multi_processor = 1
/aosp_15_r20/external/pytorch/torch/_prims/
H A Drng_prims.py71 blocks_per_sm = device_property.max_threads_per_multi_processor // block_size
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in1912 max_threads_per_multi_processor: _int
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_functions.py942 return x + prop.max_threads_per_multi_processor