diff --git a/runtime/tools/python/setup.py b/runtime/tools/python/setup.py index ea6b96f0e..ddbe3da9f 100644 --- a/runtime/tools/python/setup.py +++ b/runtime/tools/python/setup.py @@ -66,7 +66,7 @@ linklibs += ["TTRuntimeTTMetal", "tt_metal"] if enable_ttnn or enable_ttmetal: - runlibs += ["libdevice.so", "libnng.so.1", "libuv.so.1", "libfmt.so.11"] + runlibs += ["libdevice.so", "libnng.so.1", "libuv.so.1"] linklibs += ["TTRuntimeSysDesc", "TTRuntimeDebug", "TTRuntimeWorkarounds"] if enable_perf: diff --git a/test/ttmlir/Silicon/TTNN/simple_reductions.mlir b/test/ttmlir/Silicon/TTNN/simple_reductions.mlir index 9aa438ee7..908a2c67f 100644 --- a/test/ttmlir/Silicon/TTNN/simple_reductions.mlir +++ b/test/ttmlir/Silicon/TTNN/simple_reductions.mlir @@ -17,6 +17,12 @@ func.func @sum_last_2_dims(%arg0: tensor<1x32x512x64xbf16>) -> tensor<1x32xbf16> return %1 : tensor<1x32xbf16> } +func.func @sum_first_dim(%arg0: tensor<64x10xf32>) -> tensor<1x10xf32> { + %0 = tensor.empty() : tensor<1x10xf32> + %1 = "ttir.sum"(%arg0, %0) <{dim_arg = [-2 : i32], keep_dim = true, operand_constraints = [#any_device, #any_device]}> : (tensor<64x10xf32>, tensor<1x10xf32>) -> tensor<1x10xf32> + return %1: tensor<1x10xf32> +} + func.func @mean(%arg0: tensor<1x1x512x64xbf16>) -> tensor<1x1x512xbf16> { %0 = tensor.empty() : tensor<1x1x512xbf16> // CHECK: %[[C:.*]] = "ttnn.mean"[[C:.*]]