summaryrefslogtreecommitdiff
path: root/tools/jit/templates
diff options
context:
space:
mode:
authorEdward Z. Yang <ezyang@mit.edu>2018-03-01 19:51:04 -0500
committerGitHub <noreply@github.com>2018-03-01 19:51:04 -0500
commit0de5443469f29226643db5329c62bf76c86a913a (patch)
treef78bf1d119fe8f2984e4b7aff2b0b124a63a5bce /tools/jit/templates
parent27265503ad1b4c63e2caceb6b991a7abe998d7bd (diff)
downloadpytorch-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.h12
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)