[codegen] rough template for axis_info pass

2019-06-24 18:57:32 -07:00
parent 72867d17d4
commit edc31cabb0
9 changed files with 222 additions and 1 deletions
--- a/examples/python/tensorflow/dot.cpp
+++ b/examples/python/tensorflow/dot.cpp
@@ -30,7 +30,7 @@ void matmul(restrict read_only align(4) fp16 *A,
            restrict read_only align(4) fp16 *B,
            align(4) fp32 *C,
            int32 M, int32 N, int32 K,
-            int32 lda, int32 ldb, int32 ldc,
+            multiple_of(4) int32 lda, multiple_of(4) int32 ldb, multiple_of(4) int32 ldc,
            int32 *locks, int32 grid0, int32 grid1) {
  int32 rxa[TM] = get_global_range[TM](0);
  int32 ryb[TN] = get_global_range[TN](1);