Skip to content

Commit 73b58d5

Browse files
wooway777wooway777
committed
issue/1126 - fix format
1 parent 20713e1 commit 73b58d5

33 files changed

+266
-267
lines changed

‎include/infinicore/adaptor/aten_adaptor.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,9 @@
66
#include <ATen/ATen.h>
77

88
#if defined(ENABLE_NVIDIA_API) || defined(ENABLE_QY_API)
9-
#include <c10/cuda/CUDAStream.h>
10-
#include <c10/cuda/CUDAGuard.h>
119
#include <ATen/cuda/CUDAContext.h>
10+
#include <c10/cuda/CUDAGuard.h>
11+
#include <c10/cuda/CUDAStream.h>
1212
#endif
1313

1414
namespace infinicore::adaptor {

‎include/infinicore/nn/embedding.hpp‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#pragma once
22

3-
#include "module.hpp"
43
#include "../ops.hpp"
4+
#include "module.hpp"
55
#include <optional>
66

77
namespace infinicore::nn {
@@ -78,10 +78,10 @@ class Embedding : public Module {
7878
INFINICORE_NN_PARAMETER(weight);
7979

8080
private:
81-
size_t num_embeddings_; // Vocabulary size
82-
size_t embedding_dim_; // Embedding dimension
83-
std::optional<int64_t> padding_idx_; // Optional padding index
84-
DataType dtype_; // Data type for embedding weights
81+
size_t num_embeddings_; // Vocabulary size
82+
size_t embedding_dim_; // Embedding dimension
83+
std::optional<int64_t> padding_idx_; // Optional padding index
84+
DataType dtype_; // Data type for embedding weights
8585
};
8686

8787
} // namespace infinicore::nn

‎include/infinicore/nn/module.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@
33
#include "../tensor.hpp"
44
#include "parameter.hpp"
55

6+
#include <spdlog/spdlog.h>
67
#include <type_traits>
78
#include <unordered_map>
89
#include <vector>
9-
#include <spdlog/spdlog.h>
1010

1111
namespace infinicore::nn {
1212
class Module {

‎include/infinicore/ops/mha_kvcache.hpp‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@ namespace infinicore::op {
2222

2323
INFINICORE_GRAPH_OP_CLASS(
2424
MhaKVCache,
25-
Tensor, // out
26-
const Tensor &, // q
27-
const Tensor &, // k_cache
28-
const Tensor &, // v_cache
29-
const Tensor &, // seqlens_k
30-
const Tensor &, // block_table
31-
std::optional<Tensor>, // alibi_slopes
32-
float); // scale
25+
Tensor, // out
26+
const Tensor &, // q
27+
const Tensor &, // k_cache
28+
const Tensor &, // v_cache
29+
const Tensor &, // seqlens_k
30+
const Tensor &, // block_table
31+
std::optional<Tensor>, // alibi_slopes
32+
float); // scale
3333

3434
Tensor mha_kvcache(const Tensor &q,
3535
const Tensor &k_cache,

‎include/infinicore/quantization/compressed_tensors.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ class CompressedTensors : public BaseQuantization {
99
// information and support multiple quantization schemes.
1010
public:
1111
explicit CompressedTensors(const nlohmann::json &quant_config)
12-
: BaseQuantization(quant_config) {};
12+
: BaseQuantization(quant_config){};
1313

1414
infinicore::quantization::QuantScheme
1515
get_quant_scheme() const override {

‎include/infinicore/quantization/none_quantizaiton.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ class NoneQuantization : public BaseQuantization {
99
// information and support multiple quantization schemes.
1010
public:
1111
explicit NoneQuantization(const nlohmann::json &quant_config)
12-
: BaseQuantization(quant_config) {};
12+
: BaseQuantization(quant_config){};
1313

1414
infinicore::quantization::QuantScheme
1515
get_quant_scheme() const override {

‎include/infiniop/ops/add.h‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -6,20 +6,20 @@
66
typedef struct InfiniopDescriptor *infiniopAddDescriptor_t;
77

88
__INFINI_C __export infiniStatus_t infiniopCreateAddDescriptor(infiniopHandle_t handle,
9-
infiniopAddDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t c,
11-
infiniopTensorDescriptor_t a,
12-
infiniopTensorDescriptor_t b);
9+
infiniopAddDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t c,
11+
infiniopTensorDescriptor_t a,
12+
infiniopTensorDescriptor_t b);
1313

1414
__INFINI_C __export infiniStatus_t infiniopGetAddWorkspaceSize(infiniopAddDescriptor_t desc, size_t *size);
1515

1616
__INFINI_C __export infiniStatus_t infiniopAdd(infiniopAddDescriptor_t desc,
17-
void *workspace,
18-
size_t workspace_size,
19-
void *c,
20-
const void *a,
21-
const void *b,
22-
void *stream);
17+
void *workspace,
18+
size_t workspace_size,
19+
void *c,
20+
const void *a,
21+
const void *b,
22+
void *stream);
2323

2424
__INFINI_C __export infiniStatus_t infiniopDestroyAddDescriptor(infiniopAddDescriptor_t desc);
2525

‎include/infiniop/ops/add_rms_norm.h‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -18,14 +18,14 @@ __INFINI_C __export infiniStatus_t infiniopCreateAddRMSNormDescriptor(
1818
__INFINI_C __export infiniStatus_t infiniopGetAddRMSNormWorkspaceSize(infiniopAddRMSNormDescriptor_t desc, size_t *size);
1919

2020
__INFINI_C __export infiniStatus_t infiniopAddRMSNorm(infiniopAddRMSNormDescriptor_t desc,
21-
void *workspace,
22-
size_t workspace_size,
23-
void *y,
24-
void *residual_out,
25-
const void *a,
26-
const void *b,
27-
const void *weight,
28-
void *stream);
21+
void *workspace,
22+
size_t workspace_size,
23+
void *y,
24+
void *residual_out,
25+
const void *a,
26+
const void *b,
27+
const void *weight,
28+
void *stream);
2929

3030
__INFINI_C __export infiniStatus_t infiniopDestroyAddRMSNormDescriptor(infiniopAddRMSNormDescriptor_t desc);
3131

‎include/infiniop/ops/attention.h‎

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -8,27 +8,27 @@
88
typedef struct InfiniopDescriptor *infiniopAttentionDescriptor_t;
99

1010
__INFINI_C __export infiniStatus_t infiniopCreateAttentionDescriptor(infiniopHandle_t handle,
11-
infiniopAttentionDescriptor_t *desc_ptr,
12-
infiniopTensorDescriptor_t out_desc,
13-
infiniopTensorDescriptor_t q_desc,
14-
infiniopTensorDescriptor_t k_desc,
15-
infiniopTensorDescriptor_t v_desc,
16-
infiniopTensorDescriptor_t k_cache_desc,
17-
infiniopTensorDescriptor_t v_cache_desc,
18-
size_t pos);
11+
infiniopAttentionDescriptor_t *desc_ptr,
12+
infiniopTensorDescriptor_t out_desc,
13+
infiniopTensorDescriptor_t q_desc,
14+
infiniopTensorDescriptor_t k_desc,
15+
infiniopTensorDescriptor_t v_desc,
16+
infiniopTensorDescriptor_t k_cache_desc,
17+
infiniopTensorDescriptor_t v_cache_desc,
18+
size_t pos);
1919

2020
__INFINI_C __export infiniStatus_t infiniopGetAttentionWorkspaceSize(infiniopAttentionDescriptor_t desc, size_t *size);
2121

2222
__INFINI_C __export infiniStatus_t infiniopAttention(infiniopAttentionDescriptor_t desc,
23-
void *workspace,
24-
size_t workspace_size,
25-
void *out,
26-
const void *q,
27-
const void *k,
28-
const void *v,
29-
void *k_cache,
30-
void *v_cache,
31-
void *stream);
23+
void *workspace,
24+
size_t workspace_size,
25+
void *out,
26+
const void *q,
27+
const void *k,
28+
const void *v,
29+
void *k_cache,
30+
void *v_cache,
31+
void *stream);
3232

3333
__INFINI_C __export infiniStatus_t infiniopDestroyAttentionDescriptor(infiniopAttentionDescriptor_t desc);
3434
#endif

‎include/infiniop/ops/clip.h‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -6,22 +6,22 @@
66
typedef struct InfiniopDescriptor *infiniopClipDescriptor_t;
77

88
__INFINI_C __export infiniStatus_t infiniopCreateClipDescriptor(infiniopHandle_t handle,
9-
infiniopClipDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x,
12-
infiniopTensorDescriptor_t min_val,
13-
infiniopTensorDescriptor_t max_val);
9+
infiniopClipDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x,
12+
infiniopTensorDescriptor_t min_val,
13+
infiniopTensorDescriptor_t max_val);
1414

1515
__INFINI_C __export infiniStatus_t infiniopGetClipWorkspaceSize(infiniopClipDescriptor_t desc, size_t *size);
1616

1717
__INFINI_C __export infiniStatus_t infiniopClip(infiniopClipDescriptor_t desc,
18-
void *workspace,
19-
size_t workspace_size,
20-
void *y,
21-
const void *x,
22-
const void *min_val,
23-
const void *max_val,
24-
void *stream);
18+
void *workspace,
19+
size_t workspace_size,
20+
void *y,
21+
const void *x,
22+
const void *min_val,
23+
const void *max_val,
24+
void *stream);
2525

2626
__INFINI_C __export infiniStatus_t infiniopDestroyClipDescriptor(infiniopClipDescriptor_t desc);
2727

0 commit comments

Comments
 (0)