Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
a7b7ddfb
Commit
a7b7ddfb
authored
Oct 06, 2023
by
Artur Wojcik
Browse files
Merge branch 'develop' into uif2-initial
parents
80eb522f
65c37c3d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
47 additions
and
2 deletions
+47
-2
test/py/test_gpu_async.py
test/py/test_gpu_async.py
+47
-2
No files found.
test/py/test_gpu_async.py
View file @
a7b7ddfb
#####################################################################################
#####################################################################################
# The MIT License (MIT)
# The MIT License (MIT)
#
#
# Copyright (c) 2015-202
2
Advanced Micro Devices, Inc. All rights reserved.
# Copyright (c) 2015-202
3
Advanced Micro Devices, Inc. All rights reserved.
#
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# of this software and associated documentation files (the "Software"), to deal
...
@@ -23,10 +23,55 @@
...
@@ -23,10 +23,55 @@
#####################################################################################
#####################################################################################
import
migraphx
import
migraphx
import
ctypes
import
ctypes
import
os
import
glob
def
test_conv_relu
():
def
test_conv_relu
():
hip
=
ctypes
.
cdll
.
LoadLibrary
(
"libamdhip64.so"
)
# Full path of the library is needed to fix an issue on sles
# where the library is not loaded otherwise.
# We check for the presence of library at the following paths,
# in the order listed below:
#
# 1. 'rocm_path' environment variable
# 2. /opt/rocm
# 3. /opt/rocm-*
#
# If the library is not found at any of these paths, we fall back
# to the library path being detected automatically.
library
=
"libamdhip64.so"
# Environment variable containing path to rocm
rocm_path_env_var
=
"rocm_path"
# Check for rocm_path, default to /opt/rocm if it does not exist.
rocm_path_var
=
os
.
getenv
(
rocm_path_env_var
,
default
=
"/opt/rocm"
)
# Join the paths to the library to get full path,
# e.g. /opt/rocm/lib/libamdhip64.so
library_file
=
os
.
path
.
join
(
rocm_path_var
,
"lib"
,
library
)
# Check if the library file exists at the specified path
if
os
.
path
.
exists
(
library_file
):
# Replace library name by full path to the library
library
=
library_file
else
:
# Pattern match to look for path to different
# rocm versions: /opt/rocm-*
rocm_path_pattern
=
"/opt/rocm-*/lib/libamdhip64.so"
matching_libraries
=
glob
.
glob
(
rocm_path_pattern
)
if
matching_libraries
:
# Replace library name by full path to the first
# library found.
library
=
matching_libraries
[
0
]
# Loads library either by using the full path to the
# library, if it has been detected earlier,
# or, proceeds to load the library based on the name
# of the library.
hip
=
ctypes
.
cdll
.
LoadLibrary
(
library
)
p
=
migraphx
.
parse_onnx
(
"conv_relu_maxpool_test.onnx"
)
p
=
migraphx
.
parse_onnx
(
"conv_relu_maxpool_test.onnx"
)
print
(
p
)
print
(
p
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment