Generate TPU<->CPU switch functions boilerplate #165
dlibenzi
merged
15 commits into
master
from
sketch_autogen
Added first sketch of autogenerator for XLA to ATEN tensor plumbing.
18fed59a
Change gen file exec bit.
23971f80
Refining more the generation code.
775ffe15
Full code generation with registration included.
63df52fa
Blacklist CUDA functions and handle std::vector<Tensor> returns.
0f18b081
More comments and refinements.
068057dd
Added at:: namespace.
329f9fa6
asuhan
commented
on 2019-01-27
asuhan
commented
on 2019-01-27
Find the reference parameter instead of assuming 'self'.
83f5d2ba
Fix handling of void return functions.
ae22b1af
More refinements to the code generation (includes, namespaces,...).
110d5562
Added XLA TensorImpl and created bridge APIs.
3f14ef1a
Fix function name resolution.
2db78e71
Made the generated code compile.
5cd1c0a5
Default to TPU device.
da38cd16
Fix review comments.
3ac29bc1
asuhan
approved these changes
on 2019-01-29
Login to write a write a comment.
Login via GitHub