diff --git a/sam/sim/src/tiling/tile.py b/sam/sim/src/tiling/tile.py index 59022253..5a8d5df4 100644 --- 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 index c5c51a6d..1c01ec0e 100755 --- 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"