File: array1d.cu

package info (click to toggle)
python-escript 5.6-10
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 144,304 kB
  • sloc: python: 592,074; cpp: 136,909; ansic: 18,675; javascript: 9,411; xml: 3,384; sh: 738; makefile: 207
file content (44 lines) | stat: -rw-r--r-- 1,102 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include <cusp/array1d.h>
#include <cusp/blas.h>
#include <cusp/copy.h>
#include <cusp/print.h>

// This example shows how to create views to an array container.
// Views act like containers but do not own the underlying data.
// Like pointers, views are *lightweight* objects that *reference* data.

int main(void)
{
  // define array container type
  typedef cusp::array1d<int, cusp::device_memory> Array;
  
  // define array view type
  typedef Array::view View;

  // allocate array1d container with 10 elements
  Array array(10,0);

  // create view to the first 5 elements of the array
  View first_half(array.begin(), array.begin() + 5);

  // create view to the last 5 elements of the array
  View last_half(array.begin() + 5, array.end());

  // fill the first half of the array with 1s
  cusp::blas::fill(first_half, 1);
  
  // fill the first half of the array with 2s
  cusp::blas::fill(last_half,  2);

  // print the array
  cusp::print(array);

  // copy the first half to the last half
  cusp::copy(first_half, last_half);
  
  // print the array
  cusp::print(array);

  return 0;
}