Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
jeffdaily committed Oct 11, 2024
1 parent 3bcfdbc commit 06aa704
Show file tree
Hide file tree
Showing 4 changed files with 300 additions and 305 deletions.
44 changes: 22 additions & 22 deletions onnxruntime/contrib_ops/rocm/bert/attention_impl.h
Original file line number Diff line number Diff line change
Expand Up @@ -71,28 +71,28 @@ Status LaunchConcatTensorToTensor(hipStream_t stream,
half* tensor_out);

inline hipblasStatus_t _compat_hipblas_gemm_strided_batched_ex(hipblasHandle_t handle,
hipblasOperation_t transa,
hipblasOperation_t transb,
int m,
int n,
int k,
const void* alpha,
const void* A,
hipDataType a_type,
int lda,
hipblasStride stride_A,
const void* b,
hipDataType b_type,
int ldb,
hipblasStride stride_b,
const void* beta,
void* c,
hipDataType c_type,
int ldc,
hipblasStride stride_c,
int batch_count,
hipblasComputeType_t compute_type,
hipblasGemmAlgo_t algo) {
hipblasOperation_t transa,
hipblasOperation_t transb,
int m,
int n,
int k,
const void* alpha,
const void* A,
hipDataType a_type,
int lda,
hipblasStride stride_A,
const void* b,
hipDataType b_type,
int ldb,
hipblasStride stride_b,
const void* beta,
void* c,
hipDataType c_type,
int ldc,
hipblasStride stride_c,
int batch_count,
hipblasComputeType_t compute_type,
hipblasGemmAlgo_t algo) {
return hipblasGemmStridedBatchedEx(handle,
transa,
transb,
Expand Down
2 changes: 1 addition & 1 deletion onnxruntime/core/providers/rocm/rocm_stream_handle.cc
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ RocmStream::RocmStream(hipStream_t stream,
hipblasHandle_t external_hipblas_handle) : Stream(stream, device),
own_stream_(own_flag),
cpu_allocator_(cpu_allocator),
release_cpu_buffer_on_rocm_stream_(release_cpu_buffer_on_rocm_stream) {
release_cpu_buffer_on_rocm_stream_(release_cpu_buffer_on_rocm_stream) {
if (own_flag) {
HIPBLAS_CALL_THROW(hipblasCreate(&hipblas_handle_));
HIPBLAS_CALL_THROW(hipblasSetStream(hipblas_handle_, stream));
Expand Down
Loading

0 comments on commit 06aa704

Please sign in to comment.