Resolving review comments

ROCm · Jun 17, 2024 · 42a9144 · 42a9144
1 parent 8887b6d
commit 42a9144
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/rocAL/source/api/rocal_api_augmentation.cpp b/rocAL/source/api/rocal_api_augmentation.cpp
@@ -2271,8 +2271,7 @@ rocalToDecibels(
         RocalTensorDataType op_tensor_data_type = static_cast<RocalTensorDataType>(output_datatype);
         TensorInfo output_info = input->info();
         if (op_tensor_data_type != RocalTensorDataType::FP32) {
-            WRN("Only FP32 dtype is supported for To decibels augmentation.")
-            op_tensor_data_type = RocalTensorDataType::FP32;
+            THROW("Only FP32 dtype is supported for To decibels augmentation.")
         }
         output_info.set_data_type(op_tensor_data_type);
         if (input->info().layout() == RocalTensorlayout::NFT || input->info().layout() == RocalTensorlayout::NTF) // Layout is changed when input is from spectrogram/mel filter bank

diff --git a/rocAL/source/api/rocal_api_data_loaders.cpp b/rocAL/source/api/rocal_api_data_loaders.cpp
@@ -2224,7 +2224,6 @@ rocalAudioFileSourceSingleShard(
             TensorInfo output_info = info;
             std::vector<size_t> output_dims = {context->user_batch_size(), info.dims()[1], 1};
             output_info.set_dims(output_dims);
-
             auto downmixed_output = context->master_graph->create_tensor(output_info, false);
             std::shared_ptr<DownmixNode> downmix_node = context->master_graph->add_node<DownmixNode>({output}, {downmixed_output});