mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
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
|
||
|---|---|---|
| .. | ||
| bench_gen | ||
| aot_model_compiler.cc | ||
| at_launch_benchmark.cc | ||
| benchmark_args.h | ||
| benchmark_helper.cc | ||
| benchmark_helper.h | ||
| caffe2_benchmark.cc | ||
| CMakeLists.txt | ||
| 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_gpu.cc | ||
| core_overhead_benchmark.cc | ||
| db_throughput.cc | ||
| dump_operator_names.cc | ||
| 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_mpi.cc | ||
| run_plan.cc | ||
| speed_benchmark_torch.cc | ||
| speed_benchmark.cc | ||
| split_db.cc | ||
| tsv_2_proto.cc | ||
| tutorial_blob.cc | ||
| zmq_feeder.cc | ||