#0: Uncomment tt_fatal from pad_op

tenstorrent · Dec 3, 2024 · f76667e · f76667e
1 parent 0f486cb
commit f76667e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ttnn/cpp/ttnn/operations/data_movement/pad/device/pad_op.cpp b/ttnn/cpp/ttnn/operations/data_movement/pad/device/pad_op.cpp
@@ -42,9 +42,9 @@ void Pad::validate_with_output_tensors(
             "Cannot pad tilized tensor with specified format");
     } else if (input_tensor.get_layout() == Layout::ROW_MAJOR) {
         TT_FATAL(this->output_tensor_shape[3] % 2 == 0, "RM padding requires output X dim to be a multiple of 2");
-        // TT_FATAL(
-        //     input_tensor.get_dtype() == DataType::FLOAT32 || input_tensor.get_dtype() == DataType::BFLOAT16,
-        //     "Cannot pad RM tensor with specified format");
+        TT_FATAL(
+            input_tensor.get_dtype() == DataType::FLOAT32 || input_tensor.get_dtype() == DataType::BFLOAT16,
+            "Cannot pad RM tensor with specified format");
     }
 
     if (input_tensor.is_sharded()) {