Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
O
opencv
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
opencv
Commits
e6d17406
Commit
e6d17406
authored
Jan 28, 2011
by
Alexey Spizhevoy
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
added results check into multi_gpu sample
parent
b582330b
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
19 additions
and
11 deletions
+19
-11
multi_gpu.cpp
samples/gpu/multi_gpu.cpp
+19
-11
No files found.
samples/gpu/multi_gpu.cpp
View file @
e6d17406
...
...
@@ -62,13 +62,13 @@ int main()
cuSafeCall
(
cuCtxCreate
(
&
contexts
[
0
],
0
,
device
));
CUcontext
prev_context
;
cu
CtxPopCurrent
(
&
prev_context
);
cu
SafeCall
(
cuCtxPopCurrent
(
&
prev_context
)
);
// Create context for the second GPU
cuSafeCall
(
cuDeviceGet
(
&
device
,
1
));
cuSafeCall
(
cuCtxCreate
(
&
contexts
[
1
],
1
,
device
));
cu
CtxPopCurrent
(
&
prev_context
);
cu
SafeCall
(
cuCtxPopCurrent
(
&
prev_context
)
);
// Execute calculation in two threads using two GPUs
int
devices
[]
=
{
0
,
1
};
...
...
@@ -81,35 +81,42 @@ int main()
void
Worker
::
operator
()(
int
device_id
)
const
{
cuCtxPushCurrent
(
contexts
[
device_id
]);
// Set proper context
cuSafeCall
(
cuCtxPushCurrent
(
contexts
[
device_id
]));
// Generate random matrix
Mat
src
(
1000
,
1000
,
CV_32F
);
Mat
dst
;
RNG
rng
(
0
);
rng
.
fill
(
src
,
RNG
::
UNIFORM
,
0
,
1
);
// Upload data on GPU
// CPU works
transpose
(
src
,
dst
);
GpuMat
d_src
(
src
);
GpuMat
d_dst
;
// GPU works
transpose
(
d_src
,
d_dst
);
// Deallocate here, otherwise deallocation will be performed
// Check results
bool
passed
=
norm
(
dst
-
Mat
(
d_dst
),
NORM_INF
)
<
1e-3
;
cout
<<
"GPU #"
<<
device_id
<<
": "
<<
(
passed
?
"passed"
:
"FAILED"
)
<<
endl
;
// Deallocate data here, otherwise deallocation will be performed
// after context is extracted from the stack
d_src
.
release
();
d_dst
.
release
();
CUcontext
prev_context
;
cuCtxPopCurrent
(
&
prev_context
);
cout
<<
"Device "
<<
device_id
<<
" finished
\n
"
;
cuSafeCall
(
cuCtxPopCurrent
(
&
prev_context
));
}
void
destroyContexts
()
{
cu
CtxDestroy
(
contexts
[
0
]
);
cu
CtxDestroy
(
contexts
[
1
]
);
cu
SafeCall
(
cuCtxDestroy
(
contexts
[
0
])
);
cu
SafeCall
(
cuCtxDestroy
(
contexts
[
1
])
);
}
#endif
\ No newline at end of file
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