Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
pycodegen
pystencils_autodiff
Commits
8abd3aab
Commit
8abd3aab
authored
Dec 17, 2019
by
Stephan Seitz
Browse files
Fixup
parent
19f36b03
Pipeline
#20590
passed with stage
in 5 minutes and 4 seconds
Changes
1
Pipelines
2
Hide whitespace changes
Inline
Side-by-side
tests/backends/test_torch_native_compilation.py
View file @
8abd3aab
...
...
@@ -150,7 +150,9 @@ def test_torch_native_compilation_gpu():
assert
'call_backward'
in
dir
(
torch_extension
)
@
pytest
.
mark
.
skipif
(
'CI'
in
os
.
environ
,
reason
=
"GPU too old on GITLAB CI"
)
@
pytest
.
mark
.
parametrize
(
'target'
,
(
pytest
.
param
(
'gpu'
,
marks
=
pytest
.
mark
.
skipif
(
'CI'
in
os
.
environ
,
reason
=
"GPU too old on GITLAB CI"
)),
'cpu'
))
def
test_execute_torch
(
target
):
import
pycuda.autoinit
module_name
=
"Ololol"
+
target
...
...
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