debug with akhilesh

weiya711 · Jan 17, 2024 · 016c075 · 016c075
1 parent 3b00f74
commit 016c075
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/sam/sim/src/tiling/tile.py b/sam/sim/src/tiling/tile.py
@@ -529,6 +529,7 @@ def cotile_multilevel_coo(app_str, hw_config_fname, tensors, output_dir_path, hi
                         help="If this is enabled, the 'other' tensors will have at least one nonzero value")
 
     args = parser.parse_args()
+    print("Args:", args)
 
     random.seed(args.seed)
     np.random.seed(args.seed)
@@ -557,6 +558,7 @@ def cotile_multilevel_coo(app_str, hw_config_fname, tensors, output_dir_path, hi
 
         inputCache = inputCacheTensor
         tensor_path = os.path.join(FROSTT_PATH, args.input_tensor + ".tns")
+        print("tensor_path", tensor_path)
 
         # FIXME: This is broken
         frostt_tensor = FrosttTensor(tensor_path)

diff --git a/scripts/formatting/generate_tensor_tile_formats_onyx.sh b/scripts/formatting/generate_tensor_tile_formats_onyx.sh
@@ -67,10 +67,20 @@ run_format(){
 	export FROSTT_TENSOR_PATH=$filename 
 	export FROSTT_FORMATTED_PATH=$bench_path/formatted/
 	export TENSOR_FORMAT=$tensor_format
-    echo "Tensor format: $TENSOR_FORMAT"
+	echo "FROSTT_FORMATTED_TACO_PATH: $FROSTT_FORMATTED_TACO_PATH"
+	echo "FROSTT_PATH: $FROSTT_PATH"
+	echo "FROSTT_TENSOR_PATH: $FROSTT_TENSOR_PATH"
+	echo "FROSTT_FORMATTED_PATH: $FROSTT_FORMATTED_PATH"
+	echo "TENSOR_FORMAT: $TENSOR_FORMAT"
+    echo "format: $format"
 
 	$basedir/compiler/taco/build/bin/taco-test sam.pack_$format
 
+	echo "Name: $name"
+	echo "Format: $format"
+	echo "Bench: $bench"
+	echo "Bench path: $bench_path"
+
 	python3 $basedir/scripts/formatting/datastructure_tns.py -n $name -f $format --other -b $bench -hw --output_dir $bench_path/formatted/$name
 
     echo "Done processing $name"