Skip to content

Commit

Permalink
lintrunner fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
jeffdaily committed Oct 9, 2023
1 parent 2afe507 commit ba33665
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 8 deletions.
12 changes: 6 additions & 6 deletions onnxruntime/core/providers/rocm/math/softmax.cc
Original file line number Diff line number Diff line change
Expand Up @@ -39,12 +39,12 @@ Status SoftMaxComputeHelper(
gsl::narrow_cast<int>(N), tuning_ctx);
}

#define SPECIALIZED_SOFTMAX_HELPER_IMPL(T, TOut) \
template Status SoftMaxComputeHelper<T, TOut, false>(Stream * stream, const T* input, \
const TensorShape& shape, TOut* Y, int64_t axis, \
RocmTuningContext* tuning_ctx); \
template Status SoftMaxComputeHelper<T, TOut, true>(Stream * stream, const T* input, \
const TensorShape& shape, TOut* Y, int64_t axis, \
#define SPECIALIZED_SOFTMAX_HELPER_IMPL(T, TOut) \
template Status SoftMaxComputeHelper<T, TOut, false>(Stream * stream, const T* input, \
const TensorShape& shape, TOut* Y, int64_t axis, \
RocmTuningContext* tuning_ctx); \
template Status SoftMaxComputeHelper<T, TOut, true>(Stream * stream, const T* input, \
const TensorShape& shape, TOut* Y, int64_t axis, \
RocmTuningContext* tuning_ctx);

SPECIALIZED_SOFTMAX_HELPER_IMPL(MLFloat16, float)
Expand Down
2 changes: 2 additions & 0 deletions onnxruntime/core/providers/rocm/rocm_execution_provider.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1269,6 +1269,7 @@ KernelCreateInfo BuildKernelCreateInfo<void>() {
return {};
}

// clang-format off
static Status RegisterRocmKernels(KernelRegistry& kernel_registry) {
static const BuildKernelCreateInfoFn function_table[] = {
BuildKernelCreateInfo<void>, // default entry to avoid the list become empty after ops-reducing
Expand Down Expand Up @@ -2213,6 +2214,7 @@ static Status RegisterRocmKernels(KernelRegistry& kernel_registry) {

return Status::OK();
}
// clang-format on

} // namespace rocm

Expand Down
2 changes: 0 additions & 2 deletions onnxruntime/core/providers/rocm/rocm_provider_factory.cc
Original file line number Diff line number Diff line change
Expand Up @@ -242,6 +242,4 @@ extern "C" {
ORT_API(onnxruntime::Provider*, GetProvider) {
return &onnxruntime::g_provider;
}

}

0 comments on commit ba33665

Please sign in to comment.