p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!23460 Update pangu performance. 

Merge pull request !23460 from linqingke/gpt
This commit is contained in:
i-robot 2021-09-15 09:47:31 +00:00 committed by Gitee
parent 3b286500e7 6af059dac3
commit 6aac4e42d0
4 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/frontend/parallel/auto_parallel/operator_costmodel.h
View File

@ -251,6 +251,7 @@ class GeLUCost : public SqrtCost {
// Taking account of input and output
void CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) override;
};
using FastGeLUCost = GeLUCost;
using BesselI0eCost = GeLUCost;
using BesselI1eCost = GeLUCost;
using L2NormalizeCost = GeLUCost;

1
mindspore/ccsrc/frontend/parallel/dynamic_creator.h
View File

@ -78,6 +78,7 @@ class RegisterAction {
// operator register
REGISTER(MatMulInfo);
REGISTER(GeLUInfo);
REGISTER(FastGeLUInfo);
REGISTER(VirtualDatasetInfo);
REGISTER(BatchParallelInfo);
REGISTER(TanhInfo);

8
mindspore/ccsrc/frontend/parallel/ops_info/activation_info.h
View File

@ -89,6 +89,14 @@ class GeLUInfo : public ActivationOther {
~GeLUInfo() override = default;
};
class FastGeLUInfo : public ActivationOther {
public:
FastGeLUInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape,
const PrimitiveAttrs &attrs)
: ActivationOther(name, inputs_shape, outputs_shape, attrs, std::make_shared<FastGeLUCost>()) {}
~FastGeLUInfo() override = default;
};
class TanhInfo : public ActivationOther {
public:
TanhInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape,

2
model_zoo/official/nlp/pangu_alpha/src/pangu_alpha_config.py
View File

@ -36,7 +36,7 @@ class PanguAlphaConfig:
dropout_rate=0.1,
eod_token=6,
use_past=False,
hidden_act='gelu',
hidden_act='fast_gelu',
eod_reset=True,
enable_offload=False,
parallel_config=None):