(self)
| 818 | ) |
| 819 | |
| 820 | def test_math_autogradcpu(self): |
| 821 | dispatcher = PythonDispatcher() |
| 822 | dispatcher.register(["CPU", "XLA", "Lazy", "CompositeImplicitAutograd", "AutogradCPU"]) |
| 823 | self.assertExpectedInline( |
| 824 | dispatcher.dispatchTable(), |
| 825 | '''\ |
| 826 | |
| 827 | Computed Dispatch Table |
| 828 | key kernel |
| 829 | --------------------------- |
| 830 | CPU fn_CPU [kernel] |
| 831 | XLA fn_XLA [kernel] |
| 832 | Lazy fn_Lazy [kernel] |
| 833 | FPGA fn_CompositeImplicitAutograd [math kernel] |
| 834 | AutogradOther fn_CompositeImplicitAutograd [math kernel] |
| 835 | AutogradCPU fn_AutogradCPU [kernel] |
| 836 | AutogradXLA [backend fallback] |
| 837 | AutogradLazy [backend fallback] |
| 838 | ''' |
| 839 | ) |
| 840 | self.assertExpectedInline( |
| 841 | dispatcher.registrations(), |
| 842 | '''\ |
| 843 | |
| 844 | Registered Kernels |
| 845 | key kernel |
| 846 | --------------------------- |
| 847 | CPU fn_CPU |
| 848 | XLA fn_XLA |
| 849 | Lazy fn_Lazy |
| 850 | AutogradCPU fn_AutogradCPU |
| 851 | CompositeImplicitAutograd[alias] fn_CompositeImplicitAutograd |
| 852 | ''' |
| 853 | ) |
| 854 | |
| 855 | def test_defaultbackend_autogradcpu(self): |
| 856 | dispatcher = PythonDispatcher() |
nothing calls this directly
no test coverage detected