Skip to content

Commit 5dab0bb

Browse files
Add some llm ops (#1787)
Co-authored-by: DavidGu <datong_gu@163.com>
1 parent 648cdab commit 5dab0bb

File tree

131 files changed

+5998
-42
lines changed

Some content is hidden

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

131 files changed

+5998
-42
lines changed

backends/gcu/backend/executor/cast_runner.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
#include "backend/executor/tops_compiler.h"
3030
#include "backend/utils/gcu_op_desc.h"
3131
#include "backend/utils/utils.h"
32-
#include "gcu/hlir_builder/hlir_builder.h"
32+
#include "gcu/hlir/builder/hlir_builder.h"
3333
#include "runtime/runtime.h"
3434

3535
namespace backend {

backends/gcu/backend/register/register.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ limitations under the License. */
2525

2626
#include "backend/utils/gcu_op_desc.h"
2727
#include "backend/utils/types.h"
28-
#include "gcu/hlir_builder/hlir_builder.h"
28+
#include "gcu/hlir/builder/hlir_builder.h"
2929

3030
using GcuOp = ::builder::Op;
3131
using GcuOpPtr = std::shared_ptr<GcuOp>;

backends/gcu/backend/utils/types.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ limitations under the License. */
1818
#include <unordered_map>
1919
#include <vector>
2020

21-
#include "gcu/hlir_builder/hlir_builder.h"
21+
#include "gcu/hlir/builder/hlir_builder.h"
2222
#include "paddle/phi/common/data_type.h"
2323
#include "paddle/utils/blank.h"
2424
#include "paddle/utils/variant.h"

backends/gcu/backend/utils/utils.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ limitations under the License. */
2525
#include <vector>
2626

2727
#include "backend/utils/types.h"
28-
#include "gcu/hlir_builder/hlir_builder.h"
28+
#include "gcu/hlir/builder/hlir_builder.h"
2929
#include "gcu/umd/dtu_assembler_def.h"
3030
#include "paddle/phi/common/data_type.h"
3131

backends/gcu/cmake/external/gcu.cmake

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,3 +71,4 @@ endmacro()
7171

7272
find_gcu_version("topsruntime")
7373
message(STATUS "TOPS_VERSION is ${TOPS_VERSION}")
74+
set(CUSTOM_OP_API ${CMAKE_CURRENT_SOURCE_DIR}/custom_op/api/custom_ops_api.py)

backends/gcu/custom_engine/ir_translator/translator_registry.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,6 @@ class TranslatorRegistry {
6363
common::errors::InvalidArgument(
6464
"OpTranslateFunc of %s has been registered.", op_name));
6565
translator_map_.insert({op_name, op_trans_func});
66-
std::cout << "TranslatorRegistry insert " << op_name << std::endl;
6766
}
6867

6968
OpTranslateFunc Get(const std::string &op_name) const {

backends/gcu/custom_engine/ir_translator/utils/utils.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
#include <tops/tops_ext.h>
1818

1919
#include "common/utils.h"
20-
#include "gcu/hlir_builder/hlir_builder.h"
20+
#include "gcu/hlir/builder/hlir_builder.h"
2121
#include "paddle/phi/common/data_type.h"
2222

2323
using GcuOp = ::builder::Op;

0 commit comments

Comments
 (0)