pytorch/binaries
Sicheng Stephen Jia 52175307e2 [vulkan] Allow benchmark binary to handle non-single tensor inputs/outputs for Vulkan models (#73109)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/73109

This change updates the Vulkan model runner in `speed_benchmark_torch` to be able to generate inputs for models that have input/output types other than just a single tensor. Input elements are processed depending on their type.

Test Plan: Imported from OSS

Reviewed By: mikaylagawarecki

Differential Revision: D34354839

Pulled By: SS-JIA

fbshipit-source-id: 993e55372d2664fa7eddb16146deba264727f399
(cherry picked from commit 4a140202ac)
2022-02-19 01:33:51 +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 [nnc][aot_compiler] Memory formats args to aot_compiler (#72873) 2022-02-16 18:39:31 +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_gpu.cc
core_overhead_benchmark.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 Replace AutoNonVariableTypeMode with InferenceMode in fbcode. (#55114) 2021-04-02 11:45:53 -07:00
make_cifar_db.cc [caffe2] update make_cifar_db to move the string into DB::Put() (#60692) 2021-06-25 14:02:24 -07:00
make_image_db.cc [caffe2] update make_mnist_db and make_image_db to move strings into DB::Put() (#60919) 2021-06-29 11:52:43 -07:00
make_mnist_db.cc [caffe2] update make_mnist_db and make_image_db to move strings into DB::Put() (#60919) 2021-06-29 11:52:43 -07:00
optimize_for_mobile.cc Update default ouput extension in optimize_for_mobile.cc (#45598) 2020-10-15 15:34:34 -07:00
parallel_info.cc
predictor_verifier.cc
print_core_object_sizes_gpu.cc
print_registered_core_operators.cc
record_function_benchmark.cc [PyTorch] Use plain old function pointer for RecordFunctionCallback (reapply) (#49408) 2020-12-15 19:16:01 -08:00
run_plan_mpi.cc
run_plan.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
speed_benchmark.cc
split_db.cc
tsv_2_proto.cc
tutorial_blob.cc
zmq_feeder.cc