Skip to content

Commit ddec997

Browse files
committed
Issue/887 - Refactor: consolidate binary/unary operator headers and tests into unified files
1 parent 8ca4b32 commit ddec997

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

83 files changed

+1405
-4649
lines changed

include/infiniop.h

Lines changed: 4 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -2,47 +2,29 @@
22
#define __INFINIOP_API_H__
33

44
#include "infiniop/handle.h"
5-
#include "infiniop/ops/abs.h"
6-
#include "infiniop/ops/acos.h"
7-
#include "infiniop/ops/acosh.h"
5+
// Unified headers for elementwise operators
6+
#include "infiniop/ops/unary_ops_api.h"
7+
#include "infiniop/ops/binary_ops_api.h"
8+
// Other operators
89
#include "infiniop/ops/add.h"
910
#include "infiniop/ops/add_rms_norm.h"
10-
#include "infiniop/ops/asin.h"
11-
#include "infiniop/ops/asinh.h"
12-
#include "infiniop/ops/atan.h"
13-
#include "infiniop/ops/atanh.h"
1411
#include "infiniop/ops/attention.h"
15-
#include "infiniop/ops/ceil.h"
16-
#include "infiniop/ops/cos.h"
17-
#include "infiniop/ops/cosh.h"
18-
#include "infiniop/ops/erf.h"
19-
#include "infiniop/ops/floor.h"
2012
#include "infiniop/ops/causal_softmax.h"
2113
#include "infiniop/ops/clip.h"
2214
#include "infiniop/ops/conv.h"
2315
#include "infiniop/ops/dequantize_awq.h"
24-
#include "infiniop/ops/div.h"
2516
#include "infiniop/ops/gelu.h"
2617
#include "infiniop/ops/gemm.h"
2718
#include "infiniop/ops/layer_norm.h"
28-
#include "infiniop/ops/log.h"
2919
#include "infiniop/ops/logsoftmax.h"
3020
#include "infiniop/ops/lp_norm.h"
31-
#include "infiniop/ops/max.h"
32-
#include "infiniop/ops/min.h"
3321
#include "infiniop/ops/mul.h"
34-
#include "infiniop/ops/neg.h"
3522
#include "infiniop/ops/ones.h"
3623
#include "infiniop/ops/paged_attention.h"
3724
#include "infiniop/ops/paged_attention_prefill.h"
3825
#include "infiniop/ops/paged_caching.h"
3926
#include "infiniop/ops/random_sample.h"
40-
#include "infiniop/ops/reciprocal.h"
4127
#include "infiniop/ops/rearrange.h"
42-
#include "infiniop/ops/round.h"
43-
#include "infiniop/ops/sign.h"
44-
#include "infiniop/ops/sinh.h"
45-
#include "infiniop/ops/sqrt.h"
4628
#include "infiniop/ops/relu.h"
4729
#include "infiniop/ops/rms_norm.h"
4830
#include "infiniop/ops/rope.h"
@@ -52,7 +34,6 @@
5234
#include "infiniop/ops/softplus.h"
5335
#include "infiniop/ops/sub.h"
5436
#include "infiniop/ops/swiglu.h"
55-
#include "infiniop/ops/tan.h"
5637
#include "infiniop/ops/tanh.h"
5738
#include "infiniop/ops/topkrouter.h"
5839
#include "infiniop/ops/topksoftmax.h"

include/infiniop/ops/abs.h

Lines changed: 0 additions & 8 deletions
This file was deleted.

include/infiniop/ops/acos.h

Lines changed: 0 additions & 8 deletions
This file was deleted.

include/infiniop/ops/acosh.h

Lines changed: 0 additions & 8 deletions
This file was deleted.

include/infiniop/ops/asin.h

Lines changed: 0 additions & 8 deletions
This file was deleted.

include/infiniop/ops/asinh.h

Lines changed: 0 additions & 8 deletions
This file was deleted.

include/infiniop/ops/atan.h

Lines changed: 0 additions & 8 deletions
This file was deleted.

include/infiniop/ops/atanh.h

Lines changed: 0 additions & 8 deletions
This file was deleted.
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
#ifndef __INFINIOP_BINARY_OPS_API_H__
2+
#define __INFINIOP_BINARY_OPS_API_H__
3+
4+
#include "binary_op_api.h"
5+
6+
/**
7+
* @brief Unified API declarations for all binary operators.
8+
*
9+
* This header contains API declarations for all binary operators in a single file,
10+
* eliminating the need for individual header files for each operator.
11+
*
12+
* All binary operator APIs are declared here:
13+
* - div, pow, mod, max, min
14+
*/
15+
16+
// Declare all binary operator APIs
17+
BINARY_OP_API_DECLARE(div, Div)
18+
BINARY_OP_API_DECLARE(pow, Pow)
19+
BINARY_OP_API_DECLARE(mod, Mod)
20+
BINARY_OP_API_DECLARE(max, Max)
21+
BINARY_OP_API_DECLARE(min, Min)
22+
23+
#endif // __INFINIOP_BINARY_OPS_API_H__

include/infiniop/ops/ceil.h

Lines changed: 0 additions & 8 deletions
This file was deleted.

0 commit comments

Comments
 (0)