← Back to branch summary

~ubuntu-branches/ubuntu/trusty/libthrust/trusty

« back to all changes in this revision

Viewing changes to detail/device/cuda/extern_shared_ptr.h

  • Committer: Bazaar Package Importer
  • Author(s): Andreas Beckmann
  • Date: 2011-05-28 09:32:48 UTC
  • Revision ID: james.westby@ubuntu.com-20110528093248-np3euv5sj7fw3nyv
Tags: upstream-1.4.0
Import upstream version 1.4.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
detail/device/cuda/extern_shared_ptr.h
 
1
/*
 
2
 *  Copyright 2008-2011 NVIDIA Corporation
 
3
 *
 
4
 *  Licensed under the Apache License, Version 2.0 (the "License");
 
5
 *  you may not use this file except in compliance with the License.
 
6
 *  You may obtain a copy of the License at
 
7
 *
 
8
 *      http://www.apache.org/licenses/LICENSE-2.0
 
9
 *
 
10
 *  Unless required by applicable law or agreed to in writing, software
 
11
 *  distributed under the License is distributed on an "AS IS" BASIS,
 
12
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
13
 *  See the License for the specific language governing permissions and
 
14
 *  limitations under the License.
 
15
 */
 
16
 
 
17
 
 
18
#pragma once
 
19
 
 
20
#include <thrust/detail/config.h>
 
21
 
 
22
namespace thrust
 
23
{
 
24
 
 
25
namespace detail
 
26
{
 
27
 
 
28
namespace device
 
29
{
 
30
 
 
31
namespace cuda
 
32
{
 
33
 
 
34
template<typename T>
 
35
  class extern_shared_ptr
 
36
{
 
37
// don't attempt to compile with any compiler other than nvcc
 
38
// due to use of __shared__ below
 
39
#if THRUST_DEVICE_COMPILER == THRUST_DEVICE_COMPILER_NVCC
 
40
  public:
 
41
    __device__
 
42
    inline operator T * (void)
 
43
    {
 
44
      extern __shared__ int4 smem[];
 
45
      return reinterpret_cast<T*>(smem);
 
46
    }
 
47
 
 
48
    __device__
 
49
    inline operator const T * (void) const
 
50
    {
 
51
      extern __shared__ int4 smem[];
 
52
      return reinterpret_cast<const T*>(smem);
 
53
    }
 
54
#endif // THRUST_DEVICE_COMPILER_NVCC
 
55
}; // end extern_shared_ptr
 
56
 
 
57
} // end cuda
 
58
 
 
59
} // end device
 
60
 
 
61
} // end detail
 
62
 
 
63
} // end thrust
 
64