Priya Ramani 7b55dc8340 Use kernel_func_name from aotCompiler (#66337) ...

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/66337

Right now, assembly code generated for the a given method from the model is named wrapper or func by default. The function name is then replaced with a proper kernel_func_name after target specific assembly is generated.
This PR propagates a desired kernel_func_name right from aotCompiler API so that the generated function has the needed name that doesn't need to be replaced later.

Test Plan: Imported from OSS

Reviewed By: navahgar

Differential Revision: D31514095

Pulled By: priyaramani

fbshipit-source-id: b70c8e2c733600a435cd4e8b32092d37b7bf7de5

2021-10-23 02:20:45 -07: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

Use kernel_func_name from aotCompiler (#66337)

2021-10-23 02:20:45 -07:00

at_launch_benchmark.cc

Move intraop_launch_future from Parallel.h (#64166)

2021-10-08 09:07:35 -07:00

benchmark_args.h

Back out "Revert D16469619: Add Virtual Memory and CPU percentage computation to AIBench"

2019-08-07 15:44:22 -07:00

benchmark_helper.cc

[easy]Unbreak caffe2benchmarking build (#63655)

2021-08-20 12:57:27 -07:00

benchmark_helper.h

Revert D30652629: use irange for loops

2021-10-15 15:23:10 -07:00

caffe2_benchmark.cc

Remove dead includes in caffe2/binaries

2019-11-01 10:58:42 -07:00

CMakeLists.txt

Don't build shared library for AOT Compiler (#66227)

2021-10-06 15:57:32 -07: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

Fix typos, via a Levenshtein-type corrector (#31523)

2020-01-17 16:03:19 -08:00

convert_caffe_image_db.cc

Guard all Caffe2 protobuf string serializations with CAFFE_ENFORCE (fixed reverted bug) (#12848)

2018-10-23 16:21:26 -07:00

convert_db.cc

Using c10 namespace across caffe2. (#12714)

2018-10-17 12:57:19 -07:00

convert_encoded_to_raw_leveldb.cc

Remove dead includes in caffe2/binaries

2019-11-01 10:58:42 -07:00

convert_image_to_tensor.cc

Fix typos, via a Levenshtein-type corrector (#31523)

2020-01-17 16:03:19 -08:00

core_overhead_benchmark_gpu.cc

New in StaticContext returns at::DataPtr (#12029)

2018-10-03 19:10:07 -07:00

core_overhead_benchmark.cc

Lightweight at-most-once logging for API usage (#20745)

2019-05-23 23:17:59 -07:00

db_throughput.cc

Using c10 namespace across caffe2. (#12714)

2018-10-17 12:57:19 -07:00

dump_operator_names.cc

[jit] kill script namespace (#34515)

2020-03-11 23:32:48 -07:00

inspect_gpu.cc

move flags to c10 (#12144)

2018-10-04 02:09:56 -07:00

intra_inter_benchmark.cc

Update intra_inter_benchmark (#22051)

2019-06-21 23:06:27 -07:00

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

Allow a non-OpenMP based build (#19749)

2019-05-06 19:34:48 -07:00

predictor_verifier.cc

Replace c10::guts::stuff with std::stuff (#30915)

2019-12-16 13:57:19 -08:00

print_core_object_sizes_gpu.cc

Remove dead includes in caffe2/binaries

2019-11-01 10:58:42 -07:00

print_registered_core_operators.cc

Using c10 namespace across caffe2. (#12714)

2018-10-17 12:57:19 -07:00

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

Using c10 namespace across caffe2. (#12714)

2018-10-17 12:57:19 -07:00

run_plan.cc

Using c10 namespace across caffe2. (#12714)

2018-10-17 12:57:19 -07:00

speed_benchmark_torch.cc

Update Vulkan runner in benchmark binary to handle non-tensor inputs (#66123)

2021-10-05 07:59:56 -07:00

speed_benchmark.cc

Windows shared build (#13550)

2018-11-16 12:16:28 -08:00

split_db.cc

Using c10 namespace across caffe2. (#12714)

2018-10-17 12:57:19 -07:00

tsv_2_proto.cc

fix the deeptext issue (#14005)

2018-11-15 12:13:45 -08:00

tutorial_blob.cc

IValue can store Blob (#11414)

2018-09-26 01:12:31 -07:00

zmq_feeder.cc

Remove dead includes in caffe2/binaries

2019-11-01 10:58:42 -07:00