File tree 5 files changed +7
-7
lines changed
ODLA/platforms/odla_popart
5 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -36,7 +36,7 @@ DOCKER_ID=`docker ps -aq -f name=$CONTAINER_NAME -f status=running`
36
36
gid=$( id -g ${USER} )
37
37
group=$( id -g -n ${USER} )
38
38
uid=$( id -u ${USER} )
39
- extra_mnt=" -v /opt/poplar_sdk-ubuntu_18_04-2.2.2+711-26aba6cf16 :/opt/poplar_sdk:ro"
39
+ extra_mnt=" -v /opt/poplar_sdk-ubuntu_18_04-2.3.0_774 :/opt/poplar_sdk:ro"
40
40
mkdir -p /tmp/ubuntu.cache
41
41
extra_mnt=" $extra_mnt -v /tmp/ubuntu.cache:/cache"
42
42
Original file line number Diff line number Diff line change 77
77
$image_registry/computation/halo:$image_tag \
78
78
/bin/bash -c 'source scl_source enable devtoolset-7 && cd /build && cmake -G Ninja /host/halo -DHALO_USE_TIDY_CHECK=OFF -DHALO_GEN_DOCS=OFF -DODLA_BUILD_POPART_USE_CXX11ABI=OFF -DODLA_BUILD_POPART_CUSTOM_OPS=ON -DODLA_BUILD_TRT=OFF -DODLA_BUILD_XNNPACK=OFF -DODLA_BUILD_EIGEN=OFF -DODLA_BUILD_DNNL=OFF -DHALO_BUILD_RTLIB=OFF -DHALO_USE_STATIC_PROTOBUF=ON -DCPACK_SYSTEM_NAME=centos-i686 && ninja && source /opt/poplar_sdk/poplar/enable.sh && ninja check-halo && ninja package && cp /build/*.bz2 /build/*.rpm /host/output_centos'
79
79
env :
80
- poplar_sdk : /opt/poplar_sdk-centos_7_6-2.2.2+711-26aba6cf16
80
+ poplar_sdk : /opt/poplar_sdk-centos_7_6-2.3.0_774
81
81
repo_name : heterogeneity-aware-lowering-and-optimization
82
82
image_registry : registry-intl.us-west-1.aliyuncs.com
83
83
image_tag : latest-devel-x86_64-centos7
Original file line number Diff line number Diff line change @@ -35,7 +35,7 @@ set(POPART_ROOT ${POPART_ROOT} PARENT_SCOPE)
35
35
36
36
list (APPEND CMAKE_PREFIX_PATH ${POPLAR_ROOT} )
37
37
list (APPEND CMAKE_PREFIX_PATH ${POPART_ROOT} )
38
- find_package (popart 2.2.1 ... REQUIRED CONFIG COMPONENTS popart-only REQUIRED)
38
+ find_package (popart 2.3.0 ... REQUIRED CONFIG COMPONENTS popart-only REQUIRED)
39
39
40
40
message (STATUS "Found popart, version: ${popart_VERSION} " )
41
41
Original file line number Diff line number Diff line change 15
15
# ==============================================================================
16
16
17
17
add_odla_library(custom_ops SHARED
18
- erf.cc
19
18
rsqrt.cc
20
19
postprocess.cc
21
20
attention_mask.cc
Original file line number Diff line number Diff line change @@ -59,17 +59,18 @@ static popart::OpCreator<RsqrtOp> rsqrtOpCreator(popart::OpDefinitions({
59
59
class RsqrtOpx : public popart ::popx::ElementWiseUnaryOpx {
60
60
public:
61
61
RsqrtOpx (popart::Op*, popart::popx::Devicex*);
62
- void grow (poplar ::program::Sequence&) const final ;
62
+ void grow (snap ::program::Sequence&) const final ;
63
63
};
64
64
65
65
RsqrtOpx::RsqrtOpx (popart::Op* op, popart::popx::Devicex* devicex)
66
66
: popart::popx::ElementWiseUnaryOpx(op, devicex) {
67
67
verifyOp<RsqrtOp>(op, CustomOperators::Rsqrt_1);
68
68
}
69
69
70
- void RsqrtOpx::grow (poplar ::program::Sequence & prog) const {
70
+ void RsqrtOpx::grow (snap ::program::Sequence & prog) const {
71
71
auto result = popops::map (graph ().getPoplarGraph (), popops::expr::UnaryOpType::RSQRT,
72
- getInTensor (0 ).getPoplarTensor (), prog, debugContext ());
72
+ getInTensor (0 ).getPoplarTensor (), prog.getPoplarSequence (),
73
+ debugContext ());
73
74
setOutTensor (0 , snap::Tensor{result, graph ()});
74
75
}
75
76
You can’t perform that action at this time.
0 commit comments