Skip to content

Commit d1b6c93

Browse files
authored
Merge pull request PaddlePaddle#858 from graphcore/develop_to_develop-ipu_8845a1cf6a51b4fc3c80641ce03e049d3aadc487
[AUTO PR] Pulling [develop] into [develop-ipu] [2022-07-01 03:05:17+08:00]
2 parents 52c0573 + f33763e commit d1b6c93

File tree

107 files changed

+4252
-1519
lines changed

Some content is hidden

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

107 files changed

+4252
-1519
lines changed

SECURITY.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ PaddlePaddle security team attaches great importance to the security of the fram
2424

2525
### Reporting vulnerabilities
2626

27-
We encourage responsible disclosure of security issues to PaddlePaddle and please email reports about any security issues you find to security@paddlepaddle.org.
27+
We encourage responsible disclosure of security issues to PaddlePaddle and please email reports about any security issues you find to paddle-security@baidu.com.
2828

2929

3030

SECURITY_cn.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
### 报告安全问题
2222

23-
我们鼓励向飞桨负责任地披露安全问题,请将所发现的安全问题发送电子邮件到 security@paddlepaddle.org
23+
我们鼓励向飞桨负责任地披露安全问题,请将所发现的安全问题发送电子邮件到 paddle-security@baidu.com
2424

2525
在安全团队收到邮件后将会及时与您沟通并反馈问题修复进度。
2626

paddle/fluid/eager/amp_auto_cast.h

+1
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
#pragma once
1616

1717
#include "paddle/fluid/eager/api/generated/fluid_generated/dygraph_forward_api.h"
18+
#include "paddle/fluid/eager/api/manual/fluid_manual/dygraph_forward_api.h"
1819
#include "paddle/fluid/framework/convert_utils.h"
1920

2021
namespace egr {

paddle/fluid/eager/api/CMakeLists.txt

+1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
add_subdirectory(manual)
12
add_subdirectory(utils)
23
add_subdirectory(generated)
34

Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
2+
add_subdirectory(fluid_manual)
3+
set(fluid_manual_functions
4+
${fluid_manual_functions}
5+
PARENT_SCOPE)
6+
set(fluid_manual_nodes
7+
${fluid_manual_nodes}
8+
PARENT_SCOPE)
9+
endif()
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
add_subdirectory(forwards)
2+
add_subdirectory(nodes)
3+
set(fluid_manual_functions
4+
${fluid_manual_functions}
5+
PARENT_SCOPE)
6+
set(fluid_manual_nodes
7+
${fluid_manual_nodes}
8+
PARENT_SCOPE)
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
#pragma once
16+
#include "glog/logging.h"
17+
#include "paddle/fluid/eager/autograd_meta.h"
18+
#include "paddle/fluid/eager/utils.h"
19+
#include "paddle/fluid/framework/op_registry.h"
20+
#include "paddle/fluid/imperative/tracer.h"
21+
#include "paddle/phi/api/all.h"
22+
23+
std::tuple<paddle::experimental::Tensor,
24+
paddle::experimental::Tensor,
25+
paddle::experimental::Tensor,
26+
paddle::experimental::Tensor,
27+
paddle::experimental::Tensor,
28+
paddle::experimental::Tensor,
29+
paddle::experimental::Tensor,
30+
paddle::experimental::Tensor>
31+
fused_gate_attention_dygraph_function(
32+
const paddle::experimental::Tensor& Query,
33+
const paddle::experimental::Tensor& Key,
34+
const paddle::experimental::Tensor& QueryWeight,
35+
const paddle::experimental::Tensor& KeyWeight,
36+
const paddle::experimental::Tensor& ValueWeight,
37+
const paddle::experimental::Tensor& QKVWeight,
38+
const paddle::experimental::Tensor& NonbatchedBias,
39+
const paddle::experimental::Tensor& SrcMask,
40+
const paddle::experimental::Tensor& GateWeight,
41+
const paddle::experimental::Tensor& GateBias,
42+
const paddle::experimental::Tensor& OutLinearWeight,
43+
const paddle::experimental::Tensor& OutLinearBias,
44+
const paddle::framework::AttributeMap& attr_map);
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
cc_library(
2+
fused_gate_attention_fwd_func
3+
SRCS fused_gate_attention_fwd_func.cc
4+
DEPS ${eager_deps} ${fluid_deps} ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
5+
6+
add_dependencies(fused_gate_attention_fwd_func eager_codegen)
7+
8+
set(fluid_manual_functions
9+
fused_gate_attention_fwd_func
10+
PARENT_SCOPE)

0 commit comments

Comments
 (0)