File: cudpp_plan_manager.h

package info (click to toggle)
lammps 20220106.git7586adbb6a%2Bds1-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 348,064 kB
  • sloc: cpp: 831,421; python: 24,896; xml: 14,949; f90: 10,845; ansic: 7,967; sh: 4,226; perl: 4,064; fortran: 2,424; makefile: 1,501; objc: 238; lisp: 163; csh: 16; awk: 14; tcl: 6
file content (56 lines) | stat: -rw-r--r-- 1,880 bytes parent folder | download | duplicates (2)
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
45
46
47
48
49
50
51
52
53
54
55
56
// -------------------------------------------------------------
// cuDPP -- CUDA Data Parallel Primitives library
// -------------------------------------------------------------
// $Revision: 3572$
// $Date: 2010-11-23 13:04:43 -0700 (Tue, 23 Nov 2010) $
// -------------------------------------------------------------
// This source code is distributed under the terms of license.txt
// in the root directory of this source distribution.
// -------------------------------------------------------------
#ifndef __CUDPP_PLAN_MANAGER_H__
#define __CUDPP_PLAN_MANAGER_H__

#include <map>

class CUDPPPlan;
typedef void* KernelPointer;

/** @brief Singleton manager class for CUDPPPlan objects
  *
  * This class manages all active plans in CUDPP.  It is a singleton class,
  * meaning that only one instance can exist.  It is created automatically the
  * first time AddPlan() is called, and destroyed when the last plan is removed
  * using RemovePlan().
  */
class CUDPPPlanManager
{
public:
    static CUDPPHandle AddPlan(CUDPPPlan* plan);
    static bool        RemovePlan(CUDPPHandle handle);
    static CUDPPPlan*  GetPlan(CUDPPHandle handle);

    static size_t      numCTAs(KernelPointer kernel);
    static void        computeNumCTAs(KernelPointer kernel,
                                      size_t bytesDynamicSharedMem,
                                      size_t threadsPerBlock);

protected:
    static CUDPPPlanManager* m_instance;
    std::map<CUDPPHandle, CUDPPPlan*> plans;
    std::map<void*, size_t> numCTAsTable;

private:


    //! @internal Instantiate the plan manager singleton object
    static void Instantiate();
    //! @internal Destroy the plan manager singleton object
    static void Destroy();

private:
    CUDPPPlanManager() {}
    CUDPPPlanManager(const CUDPPPlanManager&) {}
    ~CUDPPPlanManager();
};

#endif // __CUDPP_PLAN_MANAGER_H__