pytorch/test/cpp
Jacob Szwejbka 474d7ec43b [Pytorch Edge] Black Box Compatibility API (#61477)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61477

It would be nice if the compatibility api was just kinda plug and play with no care about the internals of the api at all. Thats what this diff aims to provide.

The general usage would be something like
  < On the Client >
  RuntimeCompatibilityInfo runtime_info = get_runtime_compatibility_info();

  .
  .
  .
  < On the Server >
  ModelCompatibilityInfo model_info = get_model_compatibility_info(<model_path>);
  bool compatible = is_compatible(runtime_info, model_info);

Currently RuntimeCompatibilityInfo and ModelCompatibilityInfo are exactly the same, but it seemed feasible to me that they may end up diverging as more information is added to the api (such as a min supported bytecode version being exposed from the runtime).

Test Plan: unit test and ci

Reviewed By: dhruvbird, raziel

Differential Revision: D29624080

fbshipit-source-id: 43c1ce15531f6f1a92f357f9cde4e6634e561700
2021-08-03 11:27:28 -07:00
..
api Enable Gelu fp32/bf16 in CPU path using Mkldnn implementation (#58525) 2021-08-03 06:52:23 -07:00
c10d Exclude time of waiting in queue from gloo communication prof… (#61342) 2021-07-28 22:24:26 -07:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit [Pytorch Edge] Black Box Compatibility API (#61477) 2021-08-03 11:27:28 -07:00
lite_interpreter_runtime [Pytorch Delegated Backend] Save function name in debug info (#57481) 2021-05-25 13:19:02 -07:00
rpc Move torch/lib/c10d to torch/csrc/distributed/c10d (#60543) 2021-06-24 12:38:51 -07:00
tensorexpr [nnc] Simplify i%100 to i if i is less than 100; fixed #52580 (#60693) 2021-08-02 18:38:54 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00