Skip to content

Commit c08565b

Browse files
committed
cuda:移除OpT
1 parent b722c0b commit c08565b

8 files changed

Lines changed: 27 additions & 28 deletions

File tree

excuter/op-mem-cuda/src/deepx/op/arg.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
namespace deepx::op{
77

88
template<typename T>
9-
class ArgSet : public OpT<T>{
9+
class ArgSet : public Op{
1010
public:
1111
ArgSet(){
1212
this->init("argset",dtype<T>::name(), {}, {}, false, {}, {});

excuter/op-mem-cuda/src/deepx/op/concat.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
namespace deepx::op
88
{
99
template <typename T>
10-
class Concat : public OpT<T>
10+
class Concat : public Op
1111
{
1212
public:
1313
Concat(){

excuter/op-mem-cuda/src/deepx/op/elementwise.hpp

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ namespace deepx::op
1414

1515

1616
template <typename T>
17-
class Add : public OpT<T>
17+
class Add : public Op
1818
{
1919
public:
2020
Add(){
@@ -48,7 +48,7 @@ namespace deepx::op
4848
};
4949

5050
template <typename T>
51-
class Add_scalar : public OpT<T>
51+
class Add_scalar : public Op
5252
{
5353
public:
5454
Add_scalar(){
@@ -81,7 +81,7 @@ namespace deepx::op
8181
};
8282

8383
template <typename T>
84-
class Sub : public OpT<T>
84+
class Sub : public Op
8585
{
8686
public:
8787
Sub(){
@@ -115,7 +115,7 @@ namespace deepx::op
115115
}
116116
};
117117
template <typename T>
118-
class Mul : public OpT<T>
118+
class Mul : public Op
119119
{
120120
public:
121121
Mul(){
@@ -154,7 +154,7 @@ namespace deepx::op
154154
};
155155

156156
template <typename T>
157-
class Mul_scalar : public OpT<T>
157+
class Mul_scalar : public Op
158158
{
159159
public:
160160
Mul_scalar(){
@@ -191,7 +191,7 @@ namespace deepx::op
191191
};
192192

193193
template <typename T>
194-
class Div : public OpT<T>
194+
class Div : public Op
195195
{
196196
public:
197197
Div(){
@@ -236,7 +236,7 @@ namespace deepx::op
236236

237237
//Div_scalar之所以不复用Mul_scalar,是防止b接近0时,Mul_scalar(1/b)不稳定
238238
template <typename T>
239-
class Div_scalar : public OpT<T>
239+
class Div_scalar : public Op
240240
{
241241
public:
242242
Div_scalar(){
@@ -273,7 +273,7 @@ namespace deepx::op
273273
};
274274

275275
template <typename T>
276-
class Sqrt : public OpT<T>
276+
class Sqrt : public Op
277277
{
278278
public:
279279
Sqrt(){
@@ -306,7 +306,7 @@ namespace deepx::op
306306
};
307307

308308
template <typename T>
309-
class Exp : public OpT<T>
309+
class Exp : public Op
310310
{
311311
public:
312312
Exp(){
@@ -340,7 +340,7 @@ namespace deepx::op
340340
};
341341

342342
template <typename T>
343-
class Pow : public OpT<T>
343+
class Pow : public Op
344344
{
345345
public:
346346
Pow(vector< string> args, vector< string> returns, bool require_grad = false, vector< string> args_grad = {}, vector< string> returns_grad = {}){
@@ -386,7 +386,7 @@ namespace deepx::op
386386

387387

388388
template <typename T>
389-
class Pow_scalar : public OpT<T>
389+
class Pow_scalar : public Op
390390
{
391391
public:
392392
Pow_scalar(){
@@ -427,7 +427,7 @@ namespace deepx::op
427427

428428

429429
template <typename T>
430-
class Log : public OpT<T>
430+
class Log : public Op
431431
{
432432
public:
433433
Log(vector< string> args, vector< string> returns, bool require_grad = false, vector< string> args_grad = {}, vector< string> returns_grad = {}){

excuter/op-mem-cuda/src/deepx/op/init.hpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
#include "deepx/tensorfunc/init.hpp"
66
namespace deepx::op{
77
template<typename T>
8-
class Uniform : public OpT<T>{
8+
class Uniform : public Op{
99
public:
1010
Uniform(){
1111
this->init("uniform",dtype<T>::name(), {}, {}, false, {}, {});
@@ -28,7 +28,7 @@ namespace deepx::op{
2828
};
2929

3030
template<typename T>
31-
class Constant : public OpT<T>{
31+
class Constant : public Op{
3232
public:
3333
Constant(){
3434
this->init("constant",dtype<T>::name(), {}, {}, false, {}, {});
@@ -50,7 +50,7 @@ namespace deepx::op{
5050
};
5151

5252
template<typename T>
53-
class Arange : public OpT<T>{
53+
class Arange : public Op{
5454
public:
5555
Arange(){
5656
this->init("arange",dtype<T>::name(), {}, {}, false, {}, {});

excuter/op-mem-cuda/src/deepx/op/matmul.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ namespace deepx::op
1515

1616

1717
template <typename T>
18-
class MatMul : public OpT<T>
18+
class MatMul : public Op
1919
{
2020
public:
2121
MatMul(){

excuter/op-mem-cuda/src/deepx/op/new.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
#include "deepx/tensorfunc/new.hpp"
77
namespace deepx::op{
88
template<typename T>
9-
class NewTensor : public OpT<T>{
9+
class NewTensor : public Op{
1010
public:
1111
NewTensor(){
1212
this->init("newtensor",dtype<T>::name(), {}, {}, false, {}, {});
@@ -31,7 +31,7 @@ namespace deepx::op{
3131

3232

3333
template<typename T>
34-
class DelTensor : public OpT<T>{
34+
class DelTensor : public Op{
3535
public:
3636
DelTensor(){
3737
this->init("deltensor","any", {}, {}, false, {}, {});

excuter/op-mem-cuda/src/deepx/op/reduce.hpp

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,11 @@
44
#include "deepx/tensor.hpp"
55
#include "deepx/tensorfunc/reduce.hpp"
66
#include "deepx/tensorfunc/broadcast.hpp"
7-
#include "deepx/tensorfunc/compare.hpp"
8-
7+
98
namespace deepx::op
109
{
1110
template<typename T>
12-
class Sum : public OpT<T>{
11+
class Sum : public Op{
1312
public:
1413
Sum(){
1514
this->init("sum",dtype<T>::name(), {}, {}, false, {}, {});
@@ -36,7 +35,7 @@ namespace deepx::op
3635
};
3736

3837
template<typename T>
39-
class Max : public OpT<T>{
38+
class Max : public Op{
4039
public:
4140
Max(){
4241
this->init("max",dtype<T>::name(), {}, {}, false, {}, {});
@@ -67,7 +66,7 @@ namespace deepx::op
6766
};
6867

6968
template<typename T>
70-
class Max_scalar : public OpT<T>{
69+
class Max_scalar : public Op{
7170
public:
7271
Max_scalar(){
7372
this->init("max_scalar",dtype<T>::name(), {}, {}, false, {}, {});
@@ -97,7 +96,7 @@ namespace deepx::op
9796
};
9897

9998
template<typename T>
100-
class Min : public OpT<T>{
99+
class Min : public Op{
101100
public:
102101
Min(){
103102
this->init("min",dtype<T>::name(), {}, {}, false, {}, {});
@@ -126,7 +125,7 @@ namespace deepx::op
126125
};
127126

128127
template<typename T>
129-
class Min_scalar : public OpT<T>{
128+
class Min_scalar : public Op{
130129
public:
131130
Min_scalar(){
132131
this->init("min_scalar",dtype<T>::name(), {}, {}, false, {}, {});

excuter/op-mem-cuda/src/deepx/op/transpose.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ namespace deepx::op {
99
using namespace deepx::tensorfunc;
1010

1111
template <typename T>
12-
class Transpose : public OpT<T> {
12+
class Transpose : public Op {
1313
public:
1414
Transpose() {
1515
this->init("transpose", dtype<T>::name(), {}, {}, false, {}, {});

0 commit comments

Comments
 (0)