(self)
| 853 | ) |
| 854 | |
| 855 | def test_defaultbackend_autogradcpu(self): |
| 856 | dispatcher = PythonDispatcher() |
| 857 | dispatcher.register(["CPU", "XLA", "Lazy", "CompositeExplicitAutograd", "AutogradCPU"]) |
| 858 | self.assertExpectedInline( |
| 859 | dispatcher.dispatchTable(), |
| 860 | '''\ |
| 861 | |
| 862 | Computed Dispatch Table |
| 863 | key kernel |
| 864 | --------------------------- |
| 865 | CPU fn_CPU [kernel] |
| 866 | XLA fn_XLA [kernel] |
| 867 | Lazy fn_Lazy [kernel] |
| 868 | FPGA fn_CompositeExplicitAutograd [default backend kernel] |
| 869 | AutogradOther [backend fallback] |
| 870 | AutogradCPU fn_AutogradCPU [kernel] |
| 871 | AutogradXLA [backend fallback] |
| 872 | AutogradLazy [backend fallback] |
| 873 | ''' |
| 874 | ) |
| 875 | |
| 876 | self.assertExpectedInline( |
| 877 | dispatcher.registrations(), |
| 878 | '''\ |
| 879 | |
| 880 | Registered Kernels |
| 881 | key kernel |
| 882 | --------------------------- |
| 883 | CPU fn_CPU |
| 884 | XLA fn_XLA |
| 885 | Lazy fn_Lazy |
| 886 | AutogradCPU fn_AutogradCPU |
| 887 | CompositeExplicitAutograd[alias] fn_CompositeExplicitAutograd |
| 888 | ''' |
| 889 | ) |
| 890 | |
| 891 | def test_autogradother(self): |
| 892 | dispatcher = PythonDispatcher() |
nothing calls this directly
no test coverage detected