(self)
| 415 | |
| 416 | class TestComputeCodegenUnboxedKernels(unittest.TestCase): |
| 417 | def setUp(self) -> None: |
| 418 | ( |
| 419 | self.native_function_no_kern, |
| 420 | _, |
| 421 | ) = NativeFunction.from_yaml( |
| 422 | { |
| 423 | "func": "custom_1::op_1() -> bool", |
| 424 | "dispatch": {"CPU": "unused_kernel_1"}, |
| 425 | }, |
| 426 | loc=Location(__file__, 1), |
| 427 | valid_tags=set(), |
| 428 | ) |
| 429 | |
| 430 | self.default_kernel_key = ETKernelKey(default=True) |
| 431 | self.default_backend_metadata = BackendMetadata( |
| 432 | "default_kernel", False, "at::native" |
| 433 | ) |
| 434 | self.default_kernel_entry = ( |
| 435 | [self.default_kernel_key], |
| 436 | self.default_backend_metadata, |
| 437 | ) |
| 438 | |
| 439 | def test_codegen_unboxed_specialized(self) -> None: |
| 440 | specialized_kernel_key = ETKernelKey.gen_from_yaml( |
nothing calls this directly
no test coverage detected