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
9d3fb0b5
Unverified
Commit
9d3fb0b5
authored
Aug 05, 2023
by
Ted Themistokleous
Committed by
GitHub
Aug 05, 2023
Browse files
Merge branch 'develop' into enable_navi_32_ci
parents
9c91c08d
aeb9f78c
Changes
278
Hide whitespace changes
Inline
Side-by-side
Showing
18 changed files
with
342 additions
and
24 deletions
+342
-24
test/verify/CMakeLists.txt
test/verify/CMakeLists.txt
+1
-1
test/verify/run_verify.cpp
test/verify/run_verify.cpp
+23
-2
test/verify/test_add_nhwc.cpp
test/verify/test_add_nhwc.cpp
+44
-0
test/verify/test_avg_pooling_pad.cpp
test/verify/test_avg_pooling_pad.cpp
+43
-0
test/verify/test_convolution_backwards.cpp
test/verify/test_convolution_backwards.cpp
+2
-2
test/verify/test_convolution_backwards_1d.cpp
test/verify/test_convolution_backwards_1d.cpp
+2
-2
test/verify/test_convolution_backwards_2d_alt.cpp
test/verify/test_convolution_backwards_2d_alt.cpp
+47
-0
test/verify/test_convolution_backwards_2x3.cpp
test/verify/test_convolution_backwards_2x3.cpp
+2
-2
test/verify/test_convolution_backwards_3d.cpp
test/verify/test_convolution_backwards_3d.cpp
+2
-2
test/verify/test_reduce_mean_nhwc.cpp
test/verify/test_reduce_mean_nhwc.cpp
+46
-0
tools/api.py
tools/api.py
+4
-2
tools/api/api.cpp
tools/api/api.cpp
+1
-1
tools/api/migraphx.h
tools/api/migraphx.h
+3
-0
tools/format.py
tools/format.py
+108
-0
tools/generate.sh
tools/generate.sh
+3
-2
tools/include/operation.hpp
tools/include/operation.hpp
+3
-3
tools/include/pass.hpp
tools/include/pass.hpp
+1
-1
tools/te.py
tools/te.py
+7
-4
No files found.
test/verify/CMakeLists.txt
View file @
9d3fb0b5
...
...
@@ -22,7 +22,7 @@
# THE SOFTWARE.
#####################################################################################
file
(
GLOB VERIFY_TESTS
${
CONFIGURE_DEPENDS
}
*.cpp
)
file
(
GLOB VERIFY_TESTS CONFIGURE_DEPENDS *.cpp
)
add_executable
(
test_verify
${
VERIFY_TESTS
}
)
add_dependencies
(
tests test_verify
)
...
...
test/verify/run_verify.cpp
View file @
9d3fb0b5
...
...
@@ -88,10 +88,31 @@ inline void compile_check(migraphx::program& p,
auto
num
=
shapes
.
size
();
for
(
std
::
size_t
i
=
0
;
i
<
num
;
++
i
)
{
if
(
p
.
get_output_shapes
()[
i
].
lens
()
!=
shapes
[
i
].
lens
())
auto
output_shape
=
p
.
get_output_shapes
()[
i
];
if
(
output_shape
.
dynamic
()
and
shapes
[
i
].
dynamic
())
{
if
(
output_shape
.
dyn_dims
()
!=
shapes
[
i
].
dyn_dims
())
{
std
::
cout
<<
ss
.
str
()
<<
std
::
endl
;
throw
std
::
runtime_error
(
"Compiling program with "
+
name
+
" alters its dynamic output dimensions"
);
}
}
else
if
(
not
(
output_shape
.
dynamic
()
or
shapes
[
i
].
dynamic
()))
{
if
(
output_shape
.
lens
()
!=
shapes
[
i
].
lens
())
{
std
::
cout
<<
ss
.
str
()
<<
std
::
endl
;
throw
std
::
runtime_error
(
"Compiling program with "
+
name
+
" alters its static output dimensions"
);
}
}
else
{
std
::
cout
<<
ss
.
str
()
<<
std
::
endl
;
throw
std
::
runtime_error
(
"Compiling program with "
+
name
+
" alters its shape"
);
throw
std
::
runtime_error
(
"Compiling program with "
+
name
+
" alters its output dimensions (static shape vs dynamic shape)"
);
}
}
if
(
t
.
name
()
!=
"ref"
)
...
...
test/verify/test_add_nhwc.cpp
0 → 100644
View file @
9d3fb0b5
/*
* The MIT License (MIT)
*
* Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_add_nhwc
:
verify_program
<
test_add_nhwc
>
{
migraphx
::
program
create_program
()
const
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
auto
s
=
migraphx
::
shape
::
from_permutation
(
migraphx
::
shape
::
float_type
,
{
4
,
3
,
8
,
8
},
{
0
,
2
,
3
,
1
});
auto
x
=
mm
->
add_parameter
(
"x"
,
s
);
auto
y
=
mm
->
add_parameter
(
"y"
,
s
);
auto
add
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"add"
),
x
,
y
);
mm
->
add_return
({
add
});
return
p
;
}
};
test/verify/test_avg_pooling_pad.cpp
0 → 100644
View file @
9d3fb0b5
/*
* The MIT License (MIT)
*
* Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/op/pooling.hpp>
struct
test_avg_pooling_pad
:
verify_program
<
test_avg_pooling_pad
>
{
migraphx
::
program
create_program
()
const
{
// pooling test with nonzero padding
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
auto
input
=
mm
->
add_parameter
(
"x"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
3
,
7
}});
auto
op
=
migraphx
::
op
::
pooling
{
migraphx
::
op
::
pooling_mode
::
average
,
{
2
},
{
1
},
{
3
}};
mm
->
add_instruction
(
op
,
input
);
return
p
;
}
};
test/verify/test_
de
conv.cpp
→
test/verify/test_conv
olution_backwards
.cpp
View file @
9d3fb0b5
...
...
@@ -27,7 +27,7 @@
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_
de
conv
:
verify_program
<
test_
de
conv
>
struct
test_conv
olution_backwards
:
verify_program
<
test_conv
olution_backwards
>
{
migraphx
::
program
create_program
()
const
{
...
...
@@ -37,7 +37,7 @@ struct test_deconv : verify_program<test_deconv>
mm
->
add_parameter
(
"x"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
1
,
3
,
3
}});
auto
weights
=
mm
->
add_parameter
(
"w"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
1
,
3
,
3
}});
mm
->
add_instruction
(
migraphx
::
make_op
(
"
de
convolution"
),
input
,
weights
);
mm
->
add_instruction
(
migraphx
::
make_op
(
"convolution
_backwards
"
),
input
,
weights
);
return
p
;
}
};
test/verify/test_
de
conv_1d.cpp
→
test/verify/test_conv
olution_backwards
_1d.cpp
View file @
9d3fb0b5
...
...
@@ -27,7 +27,7 @@
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_
de
conv_1d
:
verify_program
<
test_
de
conv_1d
>
struct
test_conv
olution_backwards
_1d
:
verify_program
<
test_conv
olution_backwards
_1d
>
{
migraphx
::
program
create_program
()
const
{
...
...
@@ -38,7 +38,7 @@ struct test_deconv_1d : verify_program<test_deconv_1d>
auto
weights
=
mm
->
add_parameter
(
"w"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
1
,
3
}});
mm
->
add_instruction
(
migraphx
::
make_op
(
"
de
convolution"
,
migraphx
::
make_op
(
"convolution
_backwards
"
,
{{
"padding"
,
{
0
}},
{
"stride"
,
{
1
}},
{
"dilation"
,
{
1
}}}),
input
,
weights
);
...
...
test/verify/test_convolution_backwards_2d_alt.cpp
0 → 100644
View file @
9d3fb0b5
/*
* The MIT License (MIT)
*
* Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_convolution_backwards_2d_alt
:
verify_program
<
test_convolution_backwards_2d_alt
>
{
migraphx
::
program
create_program
()
const
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
auto
input
=
mm
->
add_parameter
(
"x"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
1
,
10
,
10
}});
auto
weights
=
mm
->
add_parameter
(
"w"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
1
,
3
,
3
}});
mm
->
add_instruction
(
migraphx
::
make_op
(
"convolution_backwards"
,
{{
"padding"
,
{
2
,
2
}},
{
"stride"
,
{
2
,
2
}},
{
"dilation"
,
{
2
,
2
}}}),
input
,
weights
);
return
p
;
}
};
test/verify/test_
de
conv_2x3.cpp
→
test/verify/test_conv
olution_backwards
_2x3.cpp
View file @
9d3fb0b5
...
...
@@ -27,7 +27,7 @@
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_
de
conv_2x3
:
verify_program
<
test_
de
conv_2x3
>
struct
test_conv
olution_backwards
_2x3
:
verify_program
<
test_conv
olution_backwards
_2x3
>
{
migraphx
::
program
create_program
()
const
{
...
...
@@ -38,7 +38,7 @@ struct test_deconv_2x3 : verify_program<test_deconv_2x3>
auto
weights
=
mm
->
add_parameter
(
"w"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
3
,
4
,
3
,
3
}});
mm
->
add_instruction
(
migraphx
::
make_op
(
"
de
convolution"
,
migraphx
::
make_op
(
"convolution
_backwards
"
,
{{
"padding"
,
{
1
,
1
}},
{
"stride"
,
{
2
,
3
}},
{
"dilation"
,
{
1
,
1
}}}),
input
,
weights
);
...
...
test/verify/test_
de
conv_3d.cpp
→
test/verify/test_conv
olution_backwards
_3d.cpp
View file @
9d3fb0b5
...
...
@@ -27,7 +27,7 @@
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_
de
conv_3d
:
verify_program
<
test_
de
conv_3d
>
struct
test_conv
olution_backwards
_3d
:
verify_program
<
test_conv
olution_backwards
_3d
>
{
migraphx
::
program
create_program
()
const
{
...
...
@@ -39,7 +39,7 @@ struct test_deconv_3d : verify_program<test_deconv_3d>
mm
->
add_parameter
(
"w"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
1
,
1
,
3
,
3
,
3
}});
mm
->
add_instruction
(
migraphx
::
make_op
(
"
de
convolution"
,
"convolution
_backwards
"
,
{{
"padding"
,
{
0
,
0
,
0
}},
{
"stride"
,
{
1
,
1
,
1
}},
{
"dilation"
,
{
1
,
1
,
1
}}}),
input
,
weights
);
...
...
test/verify/test_reduce_mean_nhwc.cpp
0 → 100644
View file @
9d3fb0b5
/*
* The MIT License (MIT)
*
* Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
#include <migraphx/instruction.hpp>
struct
test_reduce_mean_nhwc
:
verify_program
<
test_reduce_mean_nhwc
>
{
migraphx
::
program
create_program
()
const
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
auto
s
=
migraphx
::
shape
::
from_permutation
(
migraphx
::
shape
::
float_type
,
{
4
,
256
,
2
,
2
},
{
0
,
2
,
3
,
1
});
auto
x
=
mm
->
add_parameter
(
"x"
,
s
);
auto
reduce
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"reduce_mean"
,
{{
"axes"
,
{
1
}}}),
x
);
auto
abs
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"abs"
),
reduce
);
auto
sqrt
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"sqrt"
),
abs
);
mm
->
add_return
({
sqrt
});
return
p
;
};
};
tools/api.py
View file @
9d3fb0b5
...
...
@@ -36,6 +36,8 @@ error_type = ''
success_type
=
''
try_wrap
=
''
export_c_macro
=
'MIGRAPHX_C_EXPORT'
c_header_preamble
:
List
[
str
]
=
[]
c_api_body_preamble
:
List
[
str
]
=
[]
cpp_header_preamble
:
List
[
str
]
=
[]
...
...
@@ -125,7 +127,7 @@ class Type:
header_function
=
Template
(
'''
${error_type} ${name}(${params});
${export_c_macro}
${error_type} ${name}(${params});
'''
)
function_pointer_typedef
=
Template
(
'''
...
...
@@ -177,7 +179,7 @@ class CFunction:
**
kwargs
)
def
generate_header
(
self
)
->
str
:
return
self
.
substitute
(
header_function
)
return
self
.
substitute
(
header_function
,
export_c_macro
=
export_c_macro
)
def
generate_function_pointer
(
self
,
name
:
Optional
[
str
]
=
None
)
->
str
:
return
self
.
substitute
(
function_pointer_typedef
,
...
...
tools/api/api.cpp
View file @
9d3fb0b5
...
...
@@ -44,7 +44,7 @@ namespace migraphx {
static
thread_local
bool
disable_exception_catch
=
false
;
// NOLINT
extern
"C"
void
migraphx_test_private_disable_exception_catch
(
bool
b
)
extern
"C"
MIGRAPHX_C_EXPORT
void
migraphx_test_private_disable_exception_catch
(
bool
b
)
{
disable_exception_catch
=
b
;
}
...
...
tools/api/migraphx.h
View file @
9d3fb0b5
...
...
@@ -26,6 +26,9 @@
#include <stdlib.h>
#include <stdbool.h>
#include <migraphx/api/export.h>
// Add new types here
// clang-format off
#define MIGRAPHX_SHAPE_VISIT_TYPES(m) \
...
...
tools/format.py
0 → 100644
View file @
9d3fb0b5
#####################################################################################
# The MIT License (MIT)
#
# Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#####################################################################################
import
os
,
shutil
,
argparse
,
subprocess
CLANG_FORMAT_PATH
=
'/opt/rocm/llvm/bin'
def
run
(
cmd
,
**
kwargs
):
print
(
cmd
)
subprocess
.
run
(
cmd
,
shell
=
True
,
check
=
True
,
**
kwargs
)
def
eval
(
cmd
,
**
kwargs
):
return
subprocess
.
run
(
cmd
,
capture_output
=
True
,
shell
=
True
,
check
=
True
,
**
kwargs
).
stdout
.
decode
(
'utf-8'
).
strip
()
def
get_top
():
return
eval
(
"git rev-parse --show-toplevel"
)
def
get_head
():
return
eval
(
"git rev-parse --abbrev-ref HEAD"
)
def
get_merge_base
(
branch
):
head
=
get_head
()
return
eval
(
f
"git merge-base
{
branch
}
{
head
}
"
)
def
clang_format
(
against
,
apply
=
False
,
path
=
CLANG_FORMAT_PATH
):
base
=
get_merge_base
(
against
)
clang_format
=
os
.
path
.
join
(
path
,
'clang-format'
)
if
not
os
.
path
.
exists
(
clang_format
):
print
(
f
"
{
clang_format
}
not installed. Skipping format."
)
return
git_clang_format
=
os
.
path
.
join
(
path
,
'git-clang-format'
)
if
not
os
.
path
.
exists
(
git_clang_format
):
print
(
f
"
{
git_clang_format
}
not installed. Skipping format."
)
return
diff_flag
=
""
if
apply
else
"--diff"
run
(
f
"
{
git_clang_format
}
--binary
{
clang_format
}
{
diff_flag
}
{
base
}
"
)
def
get_files_changed
(
against
,
ext
=
(
'py'
)):
files
=
eval
(
f
"git diff-index --cached --name-only
{
against
}
"
,
cwd
=
get_top
()).
splitlines
()
return
(
f
for
f
in
files
if
f
.
endswith
(
ext
))
def
yapf_format
(
against
,
apply
=
False
):
if
not
shutil
.
which
(
'yapf'
):
print
(
"yapf not installed. Skipping format."
)
return
diff_flag
=
"--in-place"
if
apply
else
"--diff"
files
=
' '
.
join
(
get_files_changed
(
against
))
if
files
:
run
(
f
"yapf
{
diff_flag
}
-p
{
files
}
"
)
else
:
print
(
"No modified python files to format"
)
def
main
():
parser
=
argparse
.
ArgumentParser
()
parser
.
add_argument
(
'against'
,
default
=
'develop'
,
nargs
=
'?'
)
parser
.
add_argument
(
'-i'
,
'--in-place'
,
action
=
'store_true'
)
parser
.
add_argument
(
'-q'
,
'--quiet'
,
action
=
'store_true'
)
args
=
parser
.
parse_args
()
try
:
clang_format
(
args
.
against
,
apply
=
args
.
in_place
)
yapf_format
(
args
.
against
,
apply
=
args
.
in_place
)
except
subprocess
.
CalledProcessError
as
ex
:
if
ex
.
stdout
:
print
(
ex
.
stdout
.
decode
(
'utf-8'
))
if
ex
.
stderr
:
print
(
ex
.
stderr
.
decode
(
'utf-8'
))
if
not
args
.
quiet
:
print
(
f
"Command '
{
ex
.
cmd
}
' returned
{
ex
.
returncode
}
"
)
raise
# sys.exit(ex.returncode)
if
__name__
==
"__main__"
:
main
()
tools/generate.sh
View file @
9d3fb0b5
...
...
@@ -22,6 +22,7 @@
# THE SOFTWARE.
#####################################################################################
DIR
=
"
$(
cd
"
$(
dirname
"
${
BASH_SOURCE
[0]
}
"
)
"
&&
pwd
)
"
CLANG_FORMAT
=
/opt/rocm/llvm/bin/clang-format
SRC_DIR
=
$DIR
/../src
PYTHON
=
python3
if
type
-p
python3.6
>
/dev/null
;
then
...
...
@@ -30,10 +31,10 @@ fi
if
type
-p
python3.8
>
/dev/null
;
then
PYTHON
=
python3.8
fi
ls
-1
$DIR
/include/ | xargs
-n
1
-P
$(
nproc
)
-I
{}
-t
bash
-c
"
$PYTHON
$DIR
/te.py
$DIR
/include/{} |
clang-format-10
-style=file >
$SRC_DIR
/include/migraphx/{}"
ls
-1
$DIR
/include/ | xargs
-n
1
-P
$(
nproc
)
-I
{}
-t
bash
-c
"
$PYTHON
$DIR
/te.py
$DIR
/include/{} |
$CLANG_FORMAT
-style=file >
$SRC_DIR
/include/migraphx/{}"
function
api
{
$PYTHON
$DIR
/api.py
$SRC_DIR
/api/migraphx.py
$1
|
clang-format-10
-style
=
file
>
$2
$PYTHON
$DIR
/api.py
$SRC_DIR
/api/migraphx.py
$1
|
$CLANG_FORMAT
-style
=
file
>
$2
}
api
$DIR
/api/migraphx.h
$SRC_DIR
/api/include/migraphx/migraphx.h
...
...
tools/include/operation.hpp
View file @
9d3fb0b5
...
...
@@ -143,7 +143,7 @@ auto compute_shape_op(rank<2>, const T& x, const std::vector<shape>& inputs)
if
(
inputs
.
empty
())
MIGRAPHX_THROW
(
"At least one input is required for "
+
x
.
name
());
dependent_type
<
operation
,
T
>
y
=
x
;
normalize_attributes
(
y
,
inputs
[
0
]
.
max_lens
()
);
normalize_attributes
(
y
,
inputs
[
0
]);
return
any_cast
<
T
>
(
y
).
normalize_compute_shape
(
inputs
);
}
...
...
@@ -675,8 +675,8 @@ bool has_finalize(const T& x)
return
detail
::
has_finalize_op
(
x
);
}
void
migraphx_to_value
(
value
&
v
,
const
operation
&
op
);
void
migraphx_from_value
(
const
value
&
v
,
operation
&
op
);
MIGRAPHX_EXPORT
void
migraphx_to_value
(
value
&
v
,
const
operation
&
op
);
MIGRAPHX_EXPORT
void
migraphx_from_value
(
const
value
&
v
,
operation
&
op
);
#endif
...
...
tools/include/pass.hpp
View file @
9d3fb0b5
...
...
@@ -57,7 +57,7 @@ struct pass
#else
module
&
get_module
(
module_pass_manager
&
mpm
);
MIGRAPHX_EXPORT
module
&
get_module
(
module_pass_manager
&
mpm
);
namespace
detail
{
...
...
tools/te.py
View file @
9d3fb0b5
...
...
@@ -28,6 +28,8 @@ trivial = [
'bool'
,
'any_ptr'
]
export_macro
=
'MIGRAPHX_EXPORT'
headers
=
'''
#include <algorithm>
#include <cassert>
...
...
@@ -41,7 +43,7 @@ form = string.Template('''
#ifdef TYPE_ERASED_DECLARATION
// Type-erased interface for:
struct ${struct_name}
struct
${export_macro}
${struct_name}
{
${decl_members}
};
...
...
@@ -68,7 +70,7 @@ struct ${struct_name}
{
using std::swap;
auto * derived = this->any_cast<PrivateDetailTypeErasedT>();
if(derived and private_detail_te_handle_mem_var.u
nique()
)
if(derived and private_detail_te_handle_mem_var.u
se_count() == 1
)
{
*derived = std::forward<PrivateDetailTypeErasedT>(value);
}
...
...
@@ -179,7 +181,7 @@ private:
private_detail_te_handle_base_type & private_detail_te_get_handle ()
{
assert(private_detail_te_handle_mem_var != nullptr);
if (
not
private_detail_te_handle_mem_var.u
nique()
)
if (private_detail_te_handle_mem_var.u
se_count() > 1
)
private_detail_te_handle_mem_var = private_detail_te_handle_mem_var->clone();
return *private_detail_te_handle_mem_var;
}
...
...
@@ -395,7 +397,8 @@ def generate_form(name, members):
default_members
=
''
.
join
(
default_members
),
decl_members
=
''
.
join
(
decl_members
),
comment_members
=
'
\n
'
.
join
(
comment_members
),
struct_name
=
name
)
struct_name
=
name
,
export_macro
=
export_macro
)
def
virtual
(
name
,
returns
=
None
,
**
kwargs
):
...
...
Prev
1
…
10
11
12
13
14
Next
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