Compiler fix

.gitignore

@@ -7,6 +7,7 @@ __pycache__/
 *.swo
 *.txt
 *.out
+*.log
 
 # Files for MacOS and IDEs
 .DS_store
@@ -35,8 +36,12 @@ suitesparse-bench/
 sam-outputs/
 compiler/benchmark/
 
+# Generated folder for SAM tiling 
+tiles/
+
 # Generated SAM simulator tests
-*/sim/test/apps/test_*.py
+sam/sim/test/apps/
+sam/sim/test/apps/test_*.py
 
 *.gv
 
@@ -53,4 +58,8 @@ tmp_mat*/
 tensor_*_mode_*
 
 # Tensor files
-tiles/
+*.mtx
+*.tns
+
+# From instaling as pip package
+sam.egg-info/

compiler/sam-kernels.sh

@@ -37,6 +37,8 @@ GEN_KERNEL_NAMES=(
   mat_residual
   mat_elemadd3
   tensor3_mttkrp
+  tensor3_mttkrp_unfused1
+  tensor3_mttkrp_unfused2
   vec_spacc_simple
   mat_spacc_simple
   vec_sd_compression_WRONG
@@ -71,14 +73,16 @@ TACO_ARGS=(
   "X(i,j,k)=B(i,j,k) -f=X:sss -f=B:sss"
   "X(i,j,k)=B(i,j,k) -f=X:ddd -f=B:ddd"
   "X(i,j,k)=B(i,j,k)+C(i,j,k) -f=X:sss -f=B:sss -f=C:sss"
-  "x=B(i,j,k)*C(i,j,k) -f=B:sss -f=C:sss"
+  "x=B(i,j,k)*C(i,j,k) -f=B:sss -f=C:sss -s=reorder(i,j,k)"
   "X(i,j)=B(i,j,k)*c(k) -f=X:ss -f=B:sss -f=c:s"
   "X(i,j,k)=B(i,j,l)*C(k,l) -f=X:sss -f=B:sss -f=C:ss"
   "X(i,j)=B(i,j)*C(i,k)*D(k,j) -f=X:ss -f=B:ss -f=C:dd -f=D:dd:1,0 -s=reorder(i,j,k)"
   "x(i)=b*C(j,i)*d(j)+e*f(i) -f=x:s -f=C:ss:1,0 -f=d:s -f=f:s"
   "x(i)=b(i)-C(i,j)*d(j) -f=x:s -f=C:ss -f=b:s -f=d:s"
   "X(i,j)=B(i,j)+C(i,j)+D(i,j) -f=X:ss -f=B:ss -f=C:ss -f=D:ss"
   "X(i,j)=B(i,k,l)*C(j,k)*D(j,l) -f=X:ss -f=B:sss -f=C:ss -f=D:ss" 
+  "T(j,k,l)=C(j,k)*D(j,l) -f=T:sss -f=C:ss -f=D:ss" 
+  "X(i,j)=B(i,k,l)*T(j,k,l) -f=X:ss -f=B:sss -f=T:sss" 
   "x(j)=B(i,j) -f=x:s -f=B:ss"
   "X(j,k)=B(i,j,k) -f=X:ss -f=B:sss"
   "x(i)=b(i) -f=b:s -f=x:d"