Skip to content

Commit

Permalink
show supported data type in err msg
Browse files Browse the repository at this point in the history
  • Loading branch information
Masahiro Masuda committed Feb 16, 2021
1 parent c1a9a6d commit d3600f5
Showing 1 changed file with 18 additions and 9 deletions.
27 changes: 18 additions & 9 deletions src/runtime/contrib/rocthrust/thrust.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,8 @@ void thrust_sort_common(DLTensor* input, DLTensor* values_out, DLTensor* indices
} else if (out_dtype == "float64") {
thrust_sort<float, double>(input, values_out, indices_out, is_ascend, sort_len);
} else {
LOG(FATAL) << "Unsupported output dtype: " << out_dtype;
LOG(FATAL) << "Unsupported output dtype: " << out_dtype
<< ". Supported output dtypes are int32, int64, float32 and float64";
}
} else if (data_dtype == "float64") {
if (out_dtype == "int32") {
Expand All @@ -131,7 +132,8 @@ void thrust_sort_common(DLTensor* input, DLTensor* values_out, DLTensor* indices
} else if (out_dtype == "float64") {
thrust_sort<double, double>(input, values_out, indices_out, is_ascend, sort_len);
} else {
LOG(FATAL) << "Unsupported output dtype: " << out_dtype;
LOG(FATAL) << "Unsupported output dtype: " << out_dtype
<< ". Supported output dtypes are int32, int64, float32 and float64";
}
} else if (data_dtype == "int32") {
if (out_dtype == "int32") {
Expand All @@ -143,7 +145,8 @@ void thrust_sort_common(DLTensor* input, DLTensor* values_out, DLTensor* indices
} else if (out_dtype == "float64") {
thrust_sort<int32_t, double>(input, values_out, indices_out, is_ascend, sort_len);
} else {
LOG(FATAL) << "Unsupported output dtype: " << out_dtype;
LOG(FATAL) << "Unsupported output dtype: " << out_dtype
<< ". Supported output dtypes are int32, int64, float32 and float64";
}
} else if (data_dtype == "int64") {
if (out_dtype == "int32") {
Expand All @@ -155,10 +158,12 @@ void thrust_sort_common(DLTensor* input, DLTensor* values_out, DLTensor* indices
} else if (out_dtype == "float64") {
thrust_sort<int64_t, double>(input, values_out, indices_out, is_ascend, sort_len);
} else {
LOG(FATAL) << "Unsupported output dtype: " << out_dtype;
LOG(FATAL) << "Unsupported output dtype: " << out_dtype
<< ". Supported output dtypes are int32, int64, float32 and float64";
}
} else {
LOG(FATAL) << "Unsupported input dtype: " << data_dtype;
LOG(FATAL) << "Unsupported input dtype: " << data_dtype
<< ". Supported input dtypes are int32, int64, and float32 and float64.";
}
}

Expand Down Expand Up @@ -221,7 +226,8 @@ TVM_REGISTER_GLOBAL("tvm.contrib.thrust.stable_sort_by_key")
thrust_stable_sort_by_key<int, float>(keys_in, values_in, keys_out, values_out,
for_scatter);
} else {
LOG(FATAL) << "Unsupported value dtype: " << value_dtype;
LOG(FATAL) << "Unsupported value dtype: " << value_dtype
<< ". Supported value dtypes are int32, int64 and float32";
}
} else if (key_dtype == "int64") {
if (value_dtype == "int32") {
Expand All @@ -234,7 +240,8 @@ TVM_REGISTER_GLOBAL("tvm.contrib.thrust.stable_sort_by_key")
thrust_stable_sort_by_key<int64_t, float>(keys_in, values_in, keys_out, values_out,
for_scatter);
} else {
LOG(FATAL) << "Unsupported value dtype: " << value_dtype;
LOG(FATAL) << "Unsupported value dtype: " << value_dtype
<< ". Supported value dtypes are int32, int64 and float32";
}
} else if (key_dtype == "float32") {
if (value_dtype == "int32") {
Expand All @@ -247,10 +254,12 @@ TVM_REGISTER_GLOBAL("tvm.contrib.thrust.stable_sort_by_key")
thrust_stable_sort_by_key<float, float>(keys_in, values_in, keys_out, values_out,
for_scatter);
} else {
LOG(FATAL) << "Unsupported value dtype: " << value_dtype;
LOG(FATAL) << "Unsupported value dtype: " << value_dtype
<< ". Supported value dtypes are int32, int64 and float32";
}
} else {
LOG(FATAL) << "Unsupported key dtype: " << key_dtype;
LOG(FATAL) << "Unsupported key dtype: " << key_dtype
<< ". Supported key dtypes are int32, int64, and float32.";
}
});

Expand Down

0 comments on commit d3600f5

Please sign in to comment.