mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
Fangjun Kuang
184
mlx/primitives.h
184
mlx/primitives.h
@@ -40,7 +40,7 @@
|
||||
std::vector<std::vector<int>> output_shapes( \
|
||||
const std::vector<array>& inputs) override { \
|
||||
return {inputs[0].shape()}; \
|
||||
};
|
||||
}
|
||||
|
||||
namespace mlx::core {
|
||||
|
||||
@@ -154,7 +154,7 @@ class UnaryPrimitive : public Primitive {
|
||||
|
||||
class Abs : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Abs(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Abs(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -171,7 +171,7 @@ class Abs : public UnaryPrimitive {
|
||||
|
||||
class Add : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Add(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Add(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -189,7 +189,7 @@ class Add : public UnaryPrimitive {
|
||||
class AddMM : public UnaryPrimitive {
|
||||
public:
|
||||
explicit AddMM(Stream stream, float alpha, float beta)
|
||||
: UnaryPrimitive(stream), alpha_(alpha), beta_(beta) {};
|
||||
: UnaryPrimitive(stream), alpha_(alpha), beta_(beta) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -213,7 +213,7 @@ class AddMM : public UnaryPrimitive {
|
||||
class Arange : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Arange(Stream stream, double start, double stop, double step)
|
||||
: UnaryPrimitive(stream), start_(start), stop_(stop), step_(step) {};
|
||||
: UnaryPrimitive(stream), start_(start), stop_(stop), step_(step) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -231,7 +231,7 @@ class Arange : public UnaryPrimitive {
|
||||
|
||||
class ArcCos : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArcCos(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit ArcCos(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -248,7 +248,7 @@ class ArcCos : public UnaryPrimitive {
|
||||
|
||||
class ArcCosh : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArcCosh(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit ArcCosh(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -265,7 +265,7 @@ class ArcCosh : public UnaryPrimitive {
|
||||
|
||||
class ArcSin : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArcSin(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit ArcSin(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -282,7 +282,7 @@ class ArcSin : public UnaryPrimitive {
|
||||
|
||||
class ArcSinh : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArcSinh(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit ArcSinh(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -299,7 +299,7 @@ class ArcSinh : public UnaryPrimitive {
|
||||
|
||||
class ArcTan : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArcTan(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit ArcTan(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -316,7 +316,7 @@ class ArcTan : public UnaryPrimitive {
|
||||
|
||||
class ArcTan2 : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArcTan2(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit ArcTan2(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -333,7 +333,7 @@ class ArcTan2 : public UnaryPrimitive {
|
||||
|
||||
class ArcTanh : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArcTanh(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit ArcTanh(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -351,7 +351,7 @@ class ArcTanh : public UnaryPrimitive {
|
||||
class ArgPartition : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArgPartition(Stream stream, int kth, int axis)
|
||||
: UnaryPrimitive(stream), kth_(kth), axis_(axis) {};
|
||||
: UnaryPrimitive(stream), kth_(kth), axis_(axis) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -376,7 +376,7 @@ class ArgReduce : public UnaryPrimitive {
|
||||
};
|
||||
|
||||
explicit ArgReduce(Stream stream, ReduceType reduce_type, int axis)
|
||||
: UnaryPrimitive(stream), reduce_type_(reduce_type), axis_(axis) {};
|
||||
: UnaryPrimitive(stream), reduce_type_(reduce_type), axis_(axis) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -397,7 +397,7 @@ class ArgReduce : public UnaryPrimitive {
|
||||
class ArgSort : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ArgSort(Stream stream, int axis)
|
||||
: UnaryPrimitive(stream), axis_(axis) {};
|
||||
: UnaryPrimitive(stream), axis_(axis) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -416,7 +416,7 @@ class ArgSort : public UnaryPrimitive {
|
||||
class AsType : public UnaryPrimitive {
|
||||
public:
|
||||
explicit AsType(Stream stream, Dtype dtype)
|
||||
: UnaryPrimitive(stream), dtype_(dtype) {};
|
||||
: UnaryPrimitive(stream), dtype_(dtype) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -443,7 +443,7 @@ class AsStrided : public UnaryPrimitive {
|
||||
: UnaryPrimitive(stream),
|
||||
shape_(std::move(shape)),
|
||||
strides_(std::move(strides)),
|
||||
offset_(offset) {};
|
||||
offset_(offset) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -465,7 +465,7 @@ class BitwiseBinary : public UnaryPrimitive {
|
||||
enum Op { And, Or, Xor, LeftShift, RightShift };
|
||||
|
||||
explicit BitwiseBinary(Stream stream, Op op)
|
||||
: UnaryPrimitive(stream), op_(op) {};
|
||||
: UnaryPrimitive(stream), op_(op) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -482,7 +482,7 @@ class BitwiseBinary : public UnaryPrimitive {
|
||||
class BlockMaskedMM : public UnaryPrimitive {
|
||||
public:
|
||||
explicit BlockMaskedMM(Stream stream, int block_size)
|
||||
: UnaryPrimitive(stream), block_size_(block_size) {};
|
||||
: UnaryPrimitive(stream), block_size_(block_size) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -504,7 +504,7 @@ class BlockMaskedMM : public UnaryPrimitive {
|
||||
|
||||
class GatherMM : public UnaryPrimitive {
|
||||
public:
|
||||
explicit GatherMM(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit GatherMM(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -525,7 +525,7 @@ class GatherMM : public UnaryPrimitive {
|
||||
class Broadcast : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Broadcast(Stream stream, const std::vector<int>& shape)
|
||||
: UnaryPrimitive(stream), shape_(shape) {};
|
||||
: UnaryPrimitive(stream), shape_(shape) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -543,7 +543,7 @@ class Broadcast : public UnaryPrimitive {
|
||||
|
||||
class Ceil : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Ceil(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Ceil(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -604,7 +604,7 @@ class Compiled : public Primitive {
|
||||
class Concatenate : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Concatenate(Stream stream, int axis)
|
||||
: UnaryPrimitive(stream), axis_(axis) {};
|
||||
: UnaryPrimitive(stream), axis_(axis) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -622,7 +622,7 @@ class Concatenate : public UnaryPrimitive {
|
||||
|
||||
class Conjugate : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Conjugate(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Conjugate(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -652,7 +652,7 @@ class Convolution : public UnaryPrimitive {
|
||||
kernel_dilation_(kernel_dilation),
|
||||
input_dilation_(input_dilation),
|
||||
groups_(groups),
|
||||
flip_(flip) {};
|
||||
flip_(flip) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -679,7 +679,7 @@ class Convolution : public UnaryPrimitive {
|
||||
|
||||
class Copy : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Copy(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Copy(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -696,7 +696,7 @@ class Copy : public UnaryPrimitive {
|
||||
|
||||
class Cos : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Cos(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Cos(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -713,7 +713,7 @@ class Cos : public UnaryPrimitive {
|
||||
|
||||
class Cosh : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Cosh(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Cosh(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -784,7 +784,7 @@ class Depends : public Primitive {
|
||||
|
||||
class Divide : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Divide(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Divide(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -801,7 +801,7 @@ class Divide : public UnaryPrimitive {
|
||||
|
||||
class DivMod : public Primitive {
|
||||
public:
|
||||
explicit DivMod(Stream stream) : Primitive(stream) {};
|
||||
explicit DivMod(Stream stream) : Primitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
|
||||
override;
|
||||
@@ -815,7 +815,7 @@ class DivMod : public Primitive {
|
||||
std::vector<std::vector<int>> output_shapes(
|
||||
const std::vector<array>& inputs) override {
|
||||
return std::vector{inputs[0].shape(), inputs[0].shape()};
|
||||
};
|
||||
}
|
||||
|
||||
private:
|
||||
void eval(const std::vector<array>& inputs, std::vector<array>& outputs);
|
||||
@@ -823,7 +823,7 @@ class DivMod : public Primitive {
|
||||
|
||||
class Select : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Select(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Select(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -840,7 +840,7 @@ class Select : public UnaryPrimitive {
|
||||
|
||||
class Remainder : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Remainder(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Remainder(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -858,7 +858,7 @@ class Remainder : public UnaryPrimitive {
|
||||
class Equal : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Equal(Stream stream, bool equal_nan = false)
|
||||
: UnaryPrimitive(stream), equal_nan_(equal_nan) {};
|
||||
: UnaryPrimitive(stream), equal_nan_(equal_nan) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -883,7 +883,7 @@ class Equal : public UnaryPrimitive {
|
||||
|
||||
class Erf : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Erf(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Erf(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -900,7 +900,7 @@ class Erf : public UnaryPrimitive {
|
||||
|
||||
class ErfInv : public UnaryPrimitive {
|
||||
public:
|
||||
explicit ErfInv(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit ErfInv(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -917,7 +917,7 @@ class ErfInv : public UnaryPrimitive {
|
||||
|
||||
class Exp : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Exp(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Exp(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -934,7 +934,7 @@ class Exp : public UnaryPrimitive {
|
||||
|
||||
class Expm1 : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Expm1(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Expm1(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -955,7 +955,7 @@ class FFT : public UnaryPrimitive {
|
||||
const std::vector<size_t>& axes,
|
||||
bool inverse,
|
||||
bool real)
|
||||
: UnaryPrimitive(stream), axes_(axes), inverse_(inverse), real_(real) {};
|
||||
: UnaryPrimitive(stream), axes_(axes), inverse_(inverse), real_(real) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -976,7 +976,7 @@ class FFT : public UnaryPrimitive {
|
||||
|
||||
class Floor : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Floor(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Floor(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -993,7 +993,7 @@ class Floor : public UnaryPrimitive {
|
||||
|
||||
class Full : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Full(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Full(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1013,7 +1013,7 @@ class Gather : public UnaryPrimitive {
|
||||
Stream stream,
|
||||
const std::vector<int>& axes,
|
||||
const std::vector<int>& slice_sizes)
|
||||
: UnaryPrimitive(stream), axes_(axes), slice_sizes_(slice_sizes) {};
|
||||
: UnaryPrimitive(stream), axes_(axes), slice_sizes_(slice_sizes) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1031,7 +1031,7 @@ class Gather : public UnaryPrimitive {
|
||||
|
||||
class Greater : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Greater(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Greater(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1048,7 +1048,7 @@ class Greater : public UnaryPrimitive {
|
||||
|
||||
class GreaterEqual : public UnaryPrimitive {
|
||||
public:
|
||||
explicit GreaterEqual(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit GreaterEqual(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1065,7 +1065,7 @@ class GreaterEqual : public UnaryPrimitive {
|
||||
|
||||
class Less : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Less(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Less(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1082,7 +1082,7 @@ class Less : public UnaryPrimitive {
|
||||
|
||||
class LessEqual : public UnaryPrimitive {
|
||||
public:
|
||||
explicit LessEqual(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit LessEqual(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1107,7 +1107,7 @@ class Load : public UnaryPrimitive {
|
||||
: UnaryPrimitive(stream),
|
||||
reader_(reader),
|
||||
offset_(offset),
|
||||
swap_endianness_(swap_endianness) {};
|
||||
swap_endianness_(swap_endianness) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1126,7 +1126,7 @@ class Log : public UnaryPrimitive {
|
||||
enum Base { two, ten, e };
|
||||
|
||||
explicit Log(Stream stream, Base base)
|
||||
: UnaryPrimitive(stream), base_(base) {};
|
||||
: UnaryPrimitive(stream), base_(base) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1157,7 +1157,7 @@ class Log : public UnaryPrimitive {
|
||||
|
||||
class Log1p : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Log1p(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Log1p(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1173,7 +1173,7 @@ class Log1p : public UnaryPrimitive {
|
||||
|
||||
class LogicalNot : public UnaryPrimitive {
|
||||
public:
|
||||
explicit LogicalNot(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit LogicalNot(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1190,7 +1190,7 @@ class LogicalNot : public UnaryPrimitive {
|
||||
|
||||
class LogicalAnd : public UnaryPrimitive {
|
||||
public:
|
||||
explicit LogicalAnd(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit LogicalAnd(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1207,7 +1207,7 @@ class LogicalAnd : public UnaryPrimitive {
|
||||
|
||||
class LogicalOr : public UnaryPrimitive {
|
||||
public:
|
||||
explicit LogicalOr(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit LogicalOr(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1224,7 +1224,7 @@ class LogicalOr : public UnaryPrimitive {
|
||||
|
||||
class LogAddExp : public UnaryPrimitive {
|
||||
public:
|
||||
explicit LogAddExp(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit LogAddExp(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1241,7 +1241,7 @@ class LogAddExp : public UnaryPrimitive {
|
||||
|
||||
class Matmul : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Matmul(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Matmul(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1259,7 +1259,7 @@ class Matmul : public UnaryPrimitive {
|
||||
|
||||
class Maximum : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Maximum(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Maximum(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1276,7 +1276,7 @@ class Maximum : public UnaryPrimitive {
|
||||
|
||||
class Minimum : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Minimum(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Minimum(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1293,7 +1293,7 @@ class Minimum : public UnaryPrimitive {
|
||||
|
||||
class Multiply : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Multiply(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Multiply(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1310,7 +1310,7 @@ class Multiply : public UnaryPrimitive {
|
||||
|
||||
class Negative : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Negative(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Negative(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1327,7 +1327,7 @@ class Negative : public UnaryPrimitive {
|
||||
|
||||
class NotEqual : public UnaryPrimitive {
|
||||
public:
|
||||
explicit NotEqual(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit NotEqual(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1383,7 +1383,7 @@ class Pad : public UnaryPrimitive {
|
||||
: UnaryPrimitive(stream),
|
||||
axes_(axes),
|
||||
low_pad_size_(low_pad_size),
|
||||
high_pad_size_(high_pad_size) {};
|
||||
high_pad_size_(high_pad_size) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1404,7 +1404,7 @@ class Pad : public UnaryPrimitive {
|
||||
class Partition : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Partition(Stream stream, int kth, int axis)
|
||||
: UnaryPrimitive(stream), kth_(kth), axis_(axis) {};
|
||||
: UnaryPrimitive(stream), kth_(kth), axis_(axis) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1424,7 +1424,7 @@ class Partition : public UnaryPrimitive {
|
||||
|
||||
class Power : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Power(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Power(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1449,7 +1449,7 @@ class QuantizedMatmul : public UnaryPrimitive {
|
||||
: UnaryPrimitive(stream),
|
||||
group_size_(group_size),
|
||||
bits_(bits),
|
||||
transpose_(transpose) {};
|
||||
transpose_(transpose) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1473,7 +1473,7 @@ class GatherQMM : public UnaryPrimitive {
|
||||
: UnaryPrimitive(stream),
|
||||
group_size_(group_size),
|
||||
bits_(bits),
|
||||
transpose_(transpose) {};
|
||||
transpose_(transpose) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1494,7 +1494,7 @@ class GatherQMM : public UnaryPrimitive {
|
||||
class RandomBits : public UnaryPrimitive {
|
||||
public:
|
||||
explicit RandomBits(Stream stream, const std::vector<int>& shape, int width)
|
||||
: UnaryPrimitive(stream), shape_(shape), width_(width) {};
|
||||
: UnaryPrimitive(stream), shape_(shape), width_(width) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1513,7 +1513,7 @@ class RandomBits : public UnaryPrimitive {
|
||||
class Reshape : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Reshape(Stream stream, const std::vector<int>& shape)
|
||||
: UnaryPrimitive(stream), shape_(shape) {};
|
||||
: UnaryPrimitive(stream), shape_(shape) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1545,7 +1545,7 @@ class Reduce : public UnaryPrimitive {
|
||||
Stream stream,
|
||||
ReduceType reduce_type,
|
||||
const std::vector<int>& axes)
|
||||
: UnaryPrimitive(stream), reduce_type_(reduce_type), axes_(axes) {};
|
||||
: UnaryPrimitive(stream), reduce_type_(reduce_type), axes_(axes) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1594,7 +1594,7 @@ class Reduce : public UnaryPrimitive {
|
||||
|
||||
class Round : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Round(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Round(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1623,7 +1623,7 @@ class Scan : public UnaryPrimitive {
|
||||
reduce_type_(reduce_type),
|
||||
axis_(axis),
|
||||
reverse_(reverse),
|
||||
inclusive_(inclusive) {};
|
||||
inclusive_(inclusive) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1667,7 +1667,7 @@ class Scatter : public UnaryPrimitive {
|
||||
Stream stream,
|
||||
ReduceType reduce_type,
|
||||
const std::vector<int>& axes)
|
||||
: UnaryPrimitive(stream), reduce_type_(reduce_type), axes_(axes) {};
|
||||
: UnaryPrimitive(stream), reduce_type_(reduce_type), axes_(axes) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1702,7 +1702,7 @@ class Scatter : public UnaryPrimitive {
|
||||
|
||||
class Sigmoid : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Sigmoid(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Sigmoid(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1719,7 +1719,7 @@ class Sigmoid : public UnaryPrimitive {
|
||||
|
||||
class Sign : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Sign(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Sign(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1736,7 +1736,7 @@ class Sign : public UnaryPrimitive {
|
||||
|
||||
class Sin : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Sin(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Sin(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1753,7 +1753,7 @@ class Sin : public UnaryPrimitive {
|
||||
|
||||
class Sinh : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Sinh(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Sinh(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1778,7 +1778,7 @@ class Slice : public UnaryPrimitive {
|
||||
: UnaryPrimitive(stream),
|
||||
start_indices_(start_indices),
|
||||
end_indices_(end_indices),
|
||||
strides_(strides) {};
|
||||
strides_(strides) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1806,7 +1806,7 @@ class SliceUpdate : public UnaryPrimitive {
|
||||
: UnaryPrimitive(stream),
|
||||
start_indices_(start_indices),
|
||||
end_indices_(end_indices),
|
||||
strides_(strides) {};
|
||||
strides_(strides) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1829,7 +1829,7 @@ class SliceUpdate : public UnaryPrimitive {
|
||||
class Softmax : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Softmax(Stream stream, bool precise)
|
||||
: UnaryPrimitive(stream), precise_(precise) {};
|
||||
: UnaryPrimitive(stream), precise_(precise) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1849,7 +1849,7 @@ class Softmax : public UnaryPrimitive {
|
||||
class Sort : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Sort(Stream stream, int axis)
|
||||
: UnaryPrimitive(stream), axis_(axis) {};
|
||||
: UnaryPrimitive(stream), axis_(axis) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1869,7 +1869,7 @@ class Sort : public UnaryPrimitive {
|
||||
class Split : public Primitive {
|
||||
public:
|
||||
explicit Split(Stream stream, const std::vector<int>& indices, int axis)
|
||||
: Primitive(stream), indices_(indices), axis_(axis) {};
|
||||
: Primitive(stream), indices_(indices), axis_(axis) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
|
||||
override;
|
||||
@@ -1890,7 +1890,7 @@ class Split : public Primitive {
|
||||
|
||||
class Square : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Square(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Square(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1908,7 +1908,7 @@ class Square : public UnaryPrimitive {
|
||||
class Sqrt : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Sqrt(Stream stream, bool recip = false)
|
||||
: UnaryPrimitive(stream), recip_(recip) {};
|
||||
: UnaryPrimitive(stream), recip_(recip) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1933,7 +1933,7 @@ class Sqrt : public UnaryPrimitive {
|
||||
|
||||
class StopGradient : public UnaryPrimitive {
|
||||
public:
|
||||
explicit StopGradient(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit StopGradient(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1949,7 +1949,7 @@ class StopGradient : public UnaryPrimitive {
|
||||
|
||||
class Subtract : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Subtract(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Subtract(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1966,7 +1966,7 @@ class Subtract : public UnaryPrimitive {
|
||||
|
||||
class Tan : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Tan(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Tan(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -1983,7 +1983,7 @@ class Tan : public UnaryPrimitive {
|
||||
|
||||
class Tanh : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Tanh(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Tanh(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -2000,7 +2000,7 @@ class Tanh : public UnaryPrimitive {
|
||||
|
||||
class Uniform : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Uniform(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Uniform(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -2016,7 +2016,7 @@ class Uniform : public UnaryPrimitive {
|
||||
class View : public UnaryPrimitive {
|
||||
public:
|
||||
explicit View(Stream stream, Dtype dtype)
|
||||
: UnaryPrimitive(stream), dtype_(dtype) {};
|
||||
: UnaryPrimitive(stream), dtype_(dtype) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -2032,7 +2032,7 @@ class View : public UnaryPrimitive {
|
||||
class Transpose : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Transpose(Stream stream, const std::vector<int>& axes)
|
||||
: UnaryPrimitive(stream), axes_(axes) {};
|
||||
: UnaryPrimitive(stream), axes_(axes) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
@@ -2051,7 +2051,7 @@ class Transpose : public UnaryPrimitive {
|
||||
/* QR Factorization primitive. */
|
||||
class QRF : public Primitive {
|
||||
public:
|
||||
explicit QRF(Stream stream) : Primitive(stream) {};
|
||||
explicit QRF(Stream stream) : Primitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
|
||||
override;
|
||||
@@ -2067,7 +2067,7 @@ class QRF : public Primitive {
|
||||
/* SVD primitive. */
|
||||
class SVD : public Primitive {
|
||||
public:
|
||||
explicit SVD(Stream stream) : Primitive(stream) {};
|
||||
explicit SVD(Stream stream) : Primitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
|
||||
override;
|
||||
@@ -2084,7 +2084,7 @@ class SVD : public Primitive {
|
||||
/* Matrix inversion primitive. */
|
||||
class Inverse : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Inverse(Stream stream) : UnaryPrimitive(stream) {};
|
||||
explicit Inverse(Stream stream) : UnaryPrimitive(stream) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& output) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& output) override;
|
||||
@@ -2099,7 +2099,7 @@ class Inverse : public UnaryPrimitive {
|
||||
class Cholesky : public UnaryPrimitive {
|
||||
public:
|
||||
explicit Cholesky(Stream stream, bool upper)
|
||||
: UnaryPrimitive(stream), upper_(upper) {};
|
||||
: UnaryPrimitive(stream), upper_(upper) {}
|
||||
|
||||
void eval_cpu(const std::vector<array>& inputs, array& out) override;
|
||||
void eval_gpu(const std::vector<array>& inputs, array& out) override;
|
||||
|
||||
Reference in New Issue
Block a user