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 4a140202acb336412676ac090a38d7b93ae49898)
2022-02-19 01:33:51 +00:00
..
bench_gen
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
benchmark_args.h
benchmark_helper.cc
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
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