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
OpenDAS
apex
Commits
371633d5
Commit
371633d5
authored
Mar 08, 2019
by
Michael Carilli
Browse files
Repr for import error
parent
59e992da
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
4 additions
and
5 deletions
+4
-5
apex/amp/frontend.py
apex/amp/frontend.py
+0
-1
apex/amp/scaler.py
apex/amp/scaler.py
+1
-1
tests/L1/common/run_test.sh
tests/L1/common/run_test.sh
+3
-3
No files found.
apex/amp/frontend.py
View file @
371633d5
...
...
@@ -282,7 +282,6 @@ def initialize(
_amp_state
.
opt_properties
=
opt_levels
[
opt_level
](
_amp_state
.
opt_properties
)
maybe_print
(
"Selected optimization level {}"
.
format
(
opt_levels
[
opt_level
].
brief
),
True
)
maybe_print
(
"Defaults for this optimization level are:"
,
True
)
maybe_print
(
_amp_state
.
opt_properties
.
options
,
True
)
for
k
,
v
in
_amp_state
.
opt_properties
.
options
.
items
():
maybe_print
(
"{:22} : {}"
.
format
(
k
,
v
),
True
)
...
...
apex/amp/scaler.py
View file @
371633d5
...
...
@@ -49,7 +49,7 @@ class LossScaler(object):
"Warning: multi_tensor_applier fused unscale kernel is unavailable, "
"possibly because apex was installed without --cuda_ext --cpp_ext. "
"Using Python fallback. Original ImportError was: "
+
multi_tensor_applier
.
import_err
,
repr
(
multi_tensor_applier
.
import_err
)
,
True
)
LossScaler
.
has_fused_kernel
=
False
LossScaler
.
warned_no_fused_kernel
=
True
...
...
tests/L1/common/run_test.sh
View file @
371633d5
...
...
@@ -55,7 +55,7 @@ set -e
print_banner
"Installing Apex with --cuda_ext and --cpp_ext"
pushd
../../..
p
ython setup.py
install
--cuda
_ext
--
cpp
_ext
p
ip
install
-v
--no-cache-dir
--global-option
=
"--cpp
_ext
"
--
global-option
=
"--cuda
_ext
"
.
popd
for
opt_level
in
"
${
opt_levels
[@]
}
"
...
...
@@ -92,7 +92,7 @@ done
print_banner
"Reinstalling apex without extensions"
pushd
../../..
p
ython setup.py
install
p
ip
install
-v
--no-cache-dir
.
popd
for
opt_level
in
"
${
opt_levels
[@]
}
"
...
...
@@ -139,5 +139,5 @@ done
print_banner
"Reinstalling Apex with --cuda_ext and --cpp_ext"
pushd
../../..
p
ython setup.py
install
--cuda
_ext
--
cpp
_ext
p
ip
install
-v
--no-cache-dir
--global-option
=
"--cpp
_ext
"
--
global-option
=
"--cuda
_ext
"
.
popd
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