diff options
author | Edward Z. Yang <ezyang@mit.edu> | 2018-03-01 19:51:04 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-03-01 19:51:04 -0500 |
commit | 0de5443469f29226643db5329c62bf76c86a913a (patch) | |
tree | f78bf1d119fe8f2984e4b7aff2b0b124a63a5bce /tools/jit/templates | |
parent | 27265503ad1b4c63e2caceb6b991a7abe998d7bd (diff) | |
download | pytorch-0de5443469f29226643db5329c62bf76c86a913a.tar.gz pytorch-0de5443469f29226643db5329c62bf76c86a913a.tar.bz2 pytorch-0de5443469f29226643db5329c62bf76c86a913a.zip |
Reorganize interned strings into categories, autogen ATen strings. (#5471)
This also starts generating dispatch code for __and__ and similar
variants. I was too lazy to see if we have committed the '__and__ is
not inplace' mistake other places.
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Diffstat (limited to 'tools/jit/templates')
-rw-r--r-- | tools/jit/templates/aten_interned_strings.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tools/jit/templates/aten_interned_strings.h b/tools/jit/templates/aten_interned_strings.h new file mode 100644 index 0000000000..dabee2fcbb --- /dev/null +++ b/tools/jit/templates/aten_interned_strings.h @@ -0,0 +1,12 @@ +#pragma once + +// ${generated_comment} + +// ATen symbols correspond exactly to operators defined in ATen. Every +// symbol here corresponds exactly to an ATen operation which is defined +// in Declarations.yaml; attributes are in one-to-one correspondence with +// their ATen name. + +#define FORALL_ATEN_SYMBOLS(_) \ +${aten_symbols} +_(__ATEN_END) |