Alexander Belyaev 1ef4206318 [MLIR][KERNEL_GEN] Extend tf_framework.alloc to support input forwarding. ...

This PR changes the op definition and the lowering to LLVM including `tf_framework_c_interface`.
The lowering is tested with e2e test in `tf_framework_external_calls.mlir`.
The old `tf_framework_external_calls.mlir` was renamed to `tf_framework_embed_and_call.mlir` because it tests two passes at once.

PiperOrigin-RevId: 338256141
Change-Id: I4fd53ca30caae6064f7d6e15c36ca178f9ac8dee

2020-10-21 07:12:32 -07:00

..

ir

[MLIR][KERNEL_GEN] Extend tf_framework.alloc to support input forwarding.

2020-10-21 07:12:32 -07:00

tests

[MLIR][KERNEL_GEN] Extend tf_framework.alloc to support input forwarding.

2020-10-21 07:12:32 -07:00

tools/kernel-gen-opt

Refactor kernel-gen-opt.cc and remove dependency on global dialect registration (NFC)

2020-08-24 13:09:43 -07:00

transforms

[MLIR][KERNEL_GEN] Extend tf_framework.alloc to support input forwarding.

2020-10-21 07:12:32 -07:00

BUILD

Add Shape Constraint lowering in kernel_gen.

2020-10-20 04:25:27 -07:00

crash_handler.h

Print a message refering to the Kernel CodeGen tool on crashes

2020-10-19 19:27:05 -07:00

kernel_creator.cc

Add Shape Constraint lowering in kernel_gen.

2020-10-20 04:25:27 -07:00

kernel_creator.h

[MLIR][KernelGen] Support compute_xx profiles for unranked CUDA kernels

2020-10-07 04:57:42 -07:00

tf_cuda_runtime_wrappers.cc

[KERNEL_GEN] Cache cuda stream to avoid OOM.

2020-10-12 12:37:20 -07:00

tf_framework_c_interface.cc

[MLIR][KERNEL_GEN] Extend tf_framework.alloc to support input forwarding.

2020-10-21 07:12:32 -07:00

tf_framework_c_interface.h

[MLIR][KERNEL_GEN] Extend tf_framework.alloc to support input forwarding.

2020-10-21 07:12:32 -07:00

tf_to_gpu_binary.cc

Print a message refering to the Kernel CodeGen tool on crashes

2020-10-19 19:27:05 -07:00

tf_to_kernel.cc

Integrate LLVM at llvm/llvm-project@220de1f32a

2020-10-15 02:59:54 -07:00