Commit 22d1df22 authored by ExtremeViscent's avatar ExtremeViscent Committed by binmakeswell
Browse files

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h (#968)

code style
parent fb5bc6cb
......@@ -13,14 +13,16 @@
#include "cublas_wrappers.h"
#include "kernels.h"
template <typename T> class FeedForward {
public:
template <typename T>
class FeedForward {
public:
struct Config {
int outputSize;
int inputSize;
std::array<int, 3> gemm_algos;
Config(int outputs, int inputs)
: outputSize(outputs), inputSize(inputs),
: outputSize(outputs),
inputSize(inputs),
gemm_algos(std::array<int, 3>({99, 99, 99})) {}
};
......@@ -61,6 +63,6 @@ public:
config_.inputSize = inputSize;
}
private:
private:
Config config_;
};
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment