Lines Matching refs:selector

5 from torchgen.selective_build.selector import (
193         selector = SelectiveBuilder.from_yaml_str(yaml_config)
194         self.assertTrue(selector.is_operator_selected_for_training("aten::add.int"))
195         self.assertTrue(selector.is_operator_selected_for_training("aten::add"))
211         selector = SelectiveBuilder.from_yaml_str(yaml_config)
213         self.assertTrue(selector.is_kernel_dtype_selected("add_kernel", "int32"))
214         self.assertTrue(selector.is_kernel_dtype_selected("add_kernel", "int8"))
215         self.assertFalse(selector.is_kernel_dtype_selected("add_kernel", "int16"))
216         self.assertFalse(selector.is_kernel_dtype_selected("add1_kernel", "int32"))
217         self.assertFalse(selector.is_kernel_dtype_selected("add_kernel", "float"))
219         self.assertTrue(selector.is_kernel_dtype_selected("add/sub_kernel", "float"))
220         self.assertTrue(selector.is_kernel_dtype_selected("add/sub_kernel", "complex"))
221         self.assertFalse(selector.is_kernel_dtype_selected("add/sub_kernel", "int16"))
222         self.assertFalse(selector.is_kernel_dtype_selected("add/sub_kernel", "int32"))
252         selector = combine_selective_builders(selector1, selector2)
254         self.assertTrue(selector.is_kernel_dtype_selected("add_kernel", "int32"))
255         self.assertTrue(selector.is_kernel_dtype_selected("add_kernel", "int8"))
256         self.assertFalse(selector.is_kernel_dtype_selected("add_kernel", "int16"))
257         self.assertFalse(selector.is_kernel_dtype_selected("add1_kernel", "int32"))
258         self.assertFalse(selector.is_kernel_dtype_selected("add_kernel", "float"))
260         self.assertTrue(selector.is_kernel_dtype_selected("add/sub_kernel", "float"))
261         self.assertTrue(selector.is_kernel_dtype_selected("add/sub_kernel", "complex"))
262         self.assertTrue(selector.is_kernel_dtype_selected("add/sub_kernel", "none"))
263         self.assertFalse(selector.is_kernel_dtype_selected("add/sub_kernel", "int16"))
264         self.assertFalse(selector.is_kernel_dtype_selected("add/sub_kernel", "int32"))
266         self.assertTrue(selector.is_kernel_dtype_selected("mul_kernel", "int8"))
267         self.assertFalse(selector.is_kernel_dtype_selected("mul_kernel", "int32"))
274         selector = SelectiveBuilder.from_yaml_str(yaml_config)
276         self.assertTrue(selector.is_kernel_dtype_selected("add_kernel", "int32"))
277         self.assertTrue(selector.is_kernel_dtype_selected("add_kernel", "int8"))
278         self.assertTrue(selector.is_kernel_dtype_selected("add_kernel", "int16"))
279         self.assertTrue(selector.is_kernel_dtype_selected("add1_kernel", "int32"))
280         self.assertTrue(selector.is_kernel_dtype_selected("add_kernel", "float"))
294         selector = SelectiveBuilder.from_yaml_str(yaml_config)
300         self.assertTrue(selector.is_native_function_selected(native_function))
312         selector = SelectiveBuilder.from_yaml_str(yaml_config)
315             selector.et_get_selected_kernels(
326             selector.et_get_selected_kernels(
332             selector.et_get_selected_kernels(
339             selector.et_get_selected_kernels(