コード例 #1
0
ファイル: test_dispatch.py プロジェクト: zwh840282584/pytorch
 def check_invariants(actual_provenance):
     C._dispatch_check_invariants(name)
     # Normalize the test namespace so that expected outputs are stable
     actual_state = C._dispatch_dump("{}::{}".format(
         test_namespace, name)).replace(test_namespace, "test")
     actual_table = C._dispatch_dump_table("{}::{}".format(
         test_namespace, name)).replace(test_namespace, "test")
     expected_state, expected_table, expected_provenance = results.setdefault(
         frozenset(active_ops),
         Result(actual_state, actual_table, actual_provenance))
     self.assertMultiLineEqual(
         expected_state, actual_state,
         "expected from {}; actual from {}".format(
             expected_provenance, actual_provenance))
     self.assertMultiLineEqual(
         expected_table, actual_table,
         "expected from {}; actual from {}".format(
             expected_provenance, actual_provenance))
コード例 #2
0
 def rawDispatchTable(self):
     return C._dispatch_dump_table("{}::{}".format(
         self.namespace, self.name))  # type: ignore[attr-defined]