Home
last modified time | relevance | path

Searched refs:BackendFeature (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/torch/_inductor/codegen/
Dtriton.py56 BackendFeature,
2962 BackendFeature.FOREACH,
2963 BackendFeature.BUCKETIZE,
2964 BackendFeature.INPLACE_BUFFERS,
2965 BackendFeature.MASKED_SCATTER_WITH_INDEX,
2966 BackendFeature.SCAN,
2967 BackendFeature.TRITON_TEMPLATES,
2975 BackendFeature.TUPLE_REDUCTION,
2976 BackendFeature.SORT,
Dhalide.py47 BackendFeature,
1663 BackendFeature.TUPLE_REDUCTION,
1664 BackendFeature.PREFER_STORE_LOOP_ORDER,
1665 BackendFeature.REDUCE_TO_SINGLE_ELEMENT,
1669 result[BackendFeature.SCAN] = None
Dcommon.py148 class BackendFeature(Enum): class
175 assert isinstance(feature, BackendFeature)
Dcpp.py49 BackendFeature,
3896 BackendFeature.INPLACE_BUFFERS,
3897 BackendFeature.REDUCE_TO_SINGLE_ELEMENT,
/external/pytorch/torch/_inductor/kernel/
Dmm.py20 from ..codegen.common import BackendFeature
685 or not V.graph.has_feature(layout.device, BackendFeature.TRITON_TEMPLATES)
/external/pytorch/torch/_inductor/
Ddecomposition.py803 from .codegen.common import BackendFeature, has_backend_feature
805 if has_backend_feature(self.device, BackendFeature.MASKED_SCATTER_WITH_INDEX):
Dutils.py1082 from .codegen.common import BackendFeature, has_backend_feature
1092 and has_backend_feature(layout.device, BackendFeature.TRITON_TEMPLATES)
Dscheduler.py45 from .codegen.common import BackendFeature, get_scheduling_for_device, Kernel
382 and V.graph.has_feature(self.get_device(), BackendFeature.INPLACE_BUFFERS)
3594 def get_backend_features(cls, device: torch.device) -> Sequence[BackendFeature]:
Dlowering.py612 V.graph.has_feature(device, BackendFeature.FOREACH)
2051 V.graph.has_feature(input, BackendFeature.BUCKETIZE)
2052 and V.graph.has_feature(boundaries, BackendFeature.BUCKETIZE)
5914 from .codegen.common import BackendFeature, pointwise_overrides_data
/external/pytorch/torch/_inductor/fx_passes/
Dpost_grad.py24 from ..codegen.common import BackendFeature, has_backend_feature
328 and has_backend_feature("cuda", BackendFeature.TRITON_TEMPLATES)