pytorch/binaries
Priya Ramani ac97e953b4 Add dynamic shape support to AOT driver & compiler (#72995)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72995

Add ability to specify input dimensions that need to be dynamic.
Example: if dim 115 can be dynamic in input sizes "1,115;1", then specify dynamic_dims as "115"

Also recompile and update CI models and some asm code as the old ones don't compile with compiler changes in context.cpp

Test Plan: - Compiles and runs BI Bytedoc model with and without dynamic inputs.

Reviewed By: ZolotukhinM

Differential Revision: D34233121

fbshipit-source-id: 35095e549ebd6d3bec98b9abb3f0764366a0ff6f
(cherry picked from commit 33166a9f9ac9194b5df0a35280b57708df255ebd)
2022-02-24 04:30:48 +00:00
..
bench_gen Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
aot_model_compiler.cc Add dynamic shape support to AOT driver & compiler (#72995) 2022-02-24 04:30:48 +00:00
at_launch_benchmark.cc Move intraop_launch_future from Parallel.h (#64166) 2021-10-08 09:07:35 -07:00
benchmark_args.h
benchmark_helper.cc [easy]Unbreak caffe2benchmarking build (#63655) 2021-08-20 12:57:27 -07:00
benchmark_helper.h use irange for loops 2 (#66746) 2021-12-10 04:26:23 -08:00
caffe2_benchmark.cc
CMakeLists.txt [cmake] Uncomment binaries (#71157) 2022-01-12 15:01:44 -08:00
compare_models_torch.cc Small improvements to compare_models_torch binary (#65171) 2021-09-17 08:32:45 -07:00
convert_and_benchmark.cc
convert_caffe_image_db.cc
convert_db.cc
convert_encoded_to_raw_leveldb.cc
convert_image_to_tensor.cc
core_overhead_benchmark.cc
core_overhead_benchmark_gpu.cc
db_throughput.cc
dump_operator_names.cc [binaries] fix dump_operator_name binary (#71246) 2022-01-20 17:33:08 +00:00
inspect_gpu.cc
intra_inter_benchmark.cc
lite_interpreter_model_load.cc
make_cifar_db.cc
make_image_db.cc
make_mnist_db.cc
optimize_for_mobile.cc
parallel_info.cc
predictor_verifier.cc
print_core_object_sizes_gpu.cc
print_registered_core_operators.cc
record_function_benchmark.cc
run_plan.cc
run_plan_mpi.cc
speed_benchmark.cc
speed_benchmark_torch.cc [vulkan] Allow benchmark binary to handle non-single tensor inputs/outputs for Vulkan models (#73109) 2022-02-19 01:33:51 +00:00
split_db.cc
tsv_2_proto.cc
tutorial_blob.cc
zmq_feeder.cc