pytorch/test/cpp
Raghavan Raman 9ca367d48b [nnc] Use given kernel function name while emitting code (#67781)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67781

Update `LLVMCodeGen` in NNC to use the given kernel function name while emitting code.

This was earlier committed as D31445799 (c30dc52739) and got reverted as part of a stack of diffs that included a cache for `PyTorchLLVMJIT`, which was the likely culprit.

Test Plan:
```
buck test mode/opt //caffe2/test/cpp/tensorexpr:tensorexpr -- --exact 'caffe2/test/cpp/tensorexpr:tensorexpr - LLVM.CodeGenKernelFuncName'
```

Reviewed By: ZolotukhinM, bdhirsh

Differential Revision: D32145958

fbshipit-source-id: 5f4e0400c4fa7cabce5b91e6de2a294fa0cad88e
2022-01-12 15:49:17 -08:00
..
api Add flag to optionally average output attention weights across heads (#70055) 2022-01-06 17:32:37 -08:00
c10d Add support for deleteKey for FileStore (#69953) 2022-01-07 06:20:59 -08:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit Add option to load historic operators in IR when the operator is deprecated (#71148) 2022-01-12 11:07:04 -08:00
lazy [LTC] Export torch::lazy::GetBackendDevice() (#70963) 2022-01-07 13:13:18 -08:00
lite_interpreter_runtime Back out "Revert D30710710: [Pytorch Edge] Support profiling kineto events from external source" (#66421) 2021-10-12 10:55:29 -07:00
monitor torch/monitor: add pybind (#69567) 2022-01-12 13:35:11 -08:00
rpc Remove ProcessGroup from TensorPipeAgent initialization (#68128) 2021-11-11 12:28:55 -08:00
tensorexpr [nnc] Use given kernel function name while emitting code (#67781) 2022-01-12 15:49:17 -08:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00