Skip to content

Commit ecc0f05

Browse files
YqGe585duqimeng
andauthored
Update Paddle to 0116 with metax lock_runtime (#2355)
* [Metax] lock_runtime.c (#216) * update Paddle to 0116 * fix yml * update image * remote test_fused_rms_norm_op.py --------- Co-authored-by: duqimeng <[email protected]>
1 parent 94ec3be commit ecc0f05

File tree

6 files changed

+63
-702
lines changed

6 files changed

+63
-702
lines changed

.github/workflows/_IXUCA.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ jobs:
3131
if: ${{ needs.check-bypass.outputs.can-skip != 'true' && github.repository_owner == 'PaddlePaddle' }}
3232
runs-on: iluvatar-gpu-2
3333
container:
34-
image: ccr-2vdh3abv-pub.cnc.bj.baidubce.com/device/paddle-ixuca:latest
34+
image: ccr-2vdh3abv-pub.cnc.bj.baidubce.com/device/paddle-ixuca:3.3.0
3535
env:
3636
LD_LIBRARY_PATH: /usr/local/corex/lib
3737
LIBRARY_PATH: /usr/local/corex/lib
@@ -125,6 +125,7 @@ jobs:
125125
export FLAG_SKIP_FLOAT64=1
126126
cd PaddleCustomDevice/backends/iluvatar_gpu/tests
127127
# Run tests
128+
ixsmi
128129
bash run_test.sh
129130
130131
- name: Terminate and delete the container

Paddle

Submodule Paddle updated 205 files

backends/iluvatar_gpu/tests/disabled_test.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -559,3 +559,4 @@ test_tensor.py
559559
test_rms_norm.py
560560
test_api_compatibility.py
561561
test_celu_op.py
562+
test_fused_rms_norm_op.py

backends/metax_gpu/common/utils.cc

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,13 @@
1313
// limitations under the License.
1414

1515
#include "common/utils.h"
16+
#define _GNU_SOURCE
17+
#include <dlfcn.h>
18+
#include <stdio.h>
1619

1720
#include "glog/logging.h"
1821
#include "paddle/phi/backends/context_pool.h"
1922
#include "paddle/phi/backends/custom/custom_context.h"
20-
2123
namespace phi {
2224
namespace {
2325
C_Status AsyncMemCpyH2D(const C_Device device,
@@ -295,3 +297,14 @@ std::ostream& operator<<(std::ostream& os, const phi::DenseTensor& t) {
295297
#undef FOR_EACH_DATA_TYPE_TO_PRINT
296298
#undef CALL_PRINT_TENSOR
297299
} // namespace phi
300+
301+
// lock_mcruntime.c
302+
303+
__attribute__((constructor)) void lock_mcruntime() {
304+
void* handle =
305+
dlopen("/opt/maca/lib/libmcruntime.so", RTLD_LAZY | RTLD_NODELETE);
306+
if (!handle) {
307+
fprintf(stderr, "Failed to lock libmcruntime.so: %s\n", dlerror());
308+
} else {
309+
}
310+
}

0 commit comments

Comments
 (0)