Inventor/devices/SoCudaDevice.h

Go to the documentation of this file.

/*=======================================================================
 *** THE CONTENT OF THIS WORK IS PROPRIETARY TO FEI S.A.S, (FEI S.A.S.),            ***
 ***              AND IS DISTRIBUTED UNDER A LICENSE AGREEMENT.                     ***
 ***                                                                                ***
 ***  REPRODUCTION, DISCLOSURE,  OR USE,  IN WHOLE OR IN PART,  OTHER THAN AS       ***
 ***  SPECIFIED  IN THE LICENSE ARE  NOT TO BE  UNDERTAKEN  EXCEPT WITH PRIOR       ***
 ***  WRITTEN AUTHORIZATION OF FEI S.A.S.                                           ***
 ***                                                                                ***
 ***                        RESTRICTED RIGHTS LEGEND                                ***
 ***  USE, DUPLICATION, OR DISCLOSURE BY THE GOVERNMENT OF THE CONTENT OF THIS      ***
 ***  WORK OR RELATED DOCUMENTATION IS SUBJECT TO RESTRICTIONS AS SET FORTH IN      ***
 ***  SUBPARAGRAPH (C)(1) OF THE COMMERCIAL COMPUTER SOFTWARE RESTRICTED RIGHT      ***
 ***  CLAUSE  AT FAR 52.227-19  OR SUBPARAGRAPH  (C)(1)(II)  OF  THE RIGHTS IN      ***
 ***  TECHNICAL DATA AND COMPUTER SOFTWARE CLAUSE AT DFARS 52.227-7013.             ***
 ***                                                                                ***
 ***                   COPYRIGHT (C) 1996-2020 BY FEI S.A.S,                        ***
 ***                        BORDEAUX, FRANCE                                        ***
 ***                      ALL RIGHTS RESERVED                                       ***
**=======================================================================*/
/*=======================================================================
** Author      : VSG (MMM YYYY)
**=======================================================================*/
#if !defined SOCUDADEVICE_H
#define SOCUDADEVICE_H

#include <Inventor/SbVec.h>
#include <Inventor/SoDB.h>

#include <Inventor/cuda/SoCudaDefs.h>
#include <Inventor/devices/SoDevice.h>
#include <Inventor/threads/SbThreadMutex.h>


#include <Inventor/STL/vector>
#include <Inventor/STL/string>

#ifdef _WIN32
#pragma warning( push )
#pragma warning(disable:4251)
#endif

class  SoCudaDevice : public SoDevice
{
  friend class SoDevice;

  // ----------------------- Public usage --------------------------------------

public:

  static void init();

  static bool isInitialized();

  static void finish();

  static SoCudaDevice* findFirstAvailableDevice();

  static SoCudaDevice* getDevice( int index );

  static unsigned int getDevicesCount();

  virtual unsigned long long getTotalMemory() const;

  virtual unsigned long long getAvailableMemory() const;

  virtual unsigned int getLogicalUnits() const;

  virtual SbString getDriverVersion() const;

  float getComputingVersion() const;

  virtual SbString getDeviceName() const;

  int getCudaDeviceId() const;

  void getMaxGridSize( int& x, int& y, int& z ) const;

  SbVec3i32 getMaxGridSize() const;

  SbVec3i32 getMaxBlockSize() const;

  int getMaxThreads() const;

  unsigned long long getMaxSharedMemoryPerBlock() const;

  unsigned long long getMemPitch() const;

  friend std::ostream& operator << ( std::ostream& os, const SoCudaDevice& dev )
  {
    unsigned long long allocatedMemory, availableMemory;
    dev.getMemoryInfo( allocatedMemory, availableMemory );

    return os << "[GPU name]: " << dev.m_name.toLatin1() << "\n" << \
                 "[Driver ver]: " << dev.m_drvVer << "\n" << \
                 "[Vram size]: " << (dev.m_totalMemory)/(1024*1024) << "MB" << "\n" << // memory size returned in Mbytes.
                 "[Available Vram size]: " << availableMemory/(1024*1024) << "MB" << "\n" << // memory size returned in Mbytes.
                 "[LogicalUnits]: " << dev.m_logicalUnits;
  }

  // ----------------------- Protected usage --------------------------------------

private:

  SoCudaDevice();


  virtual ~SoCudaDevice();

  // ----------------------- Private usage --------------------------------------

private:

  static bool findFirstDeviceFunc( SoDevice* device );


  // ----------------------- Private usage --------------------------------------

private:

  void getMemoryInfo( unsigned long long& allocatedMemory, unsigned long long& availableMemory ) const;

  unsigned long long m_totalMemory;

  unsigned int m_logicalUnits;

  float m_computingVersion;

  // Driver version
  SbString m_drvVer;

  unsigned int m_devicePtr;

  SbString m_name;

  SbVec3i32 m_maxGridSize;

  SbVec3i32 m_maxBlockSize;

  int m_maxThreadsPerBlock;

  unsigned long long m_maxSharedMemoryPerBlock;

  unsigned long long m_memPitch;

  static std::vector< SoCudaDevice * > s_cudaDevices;

  // We keep the number of references to this module, so we really free the data when nobody uses it..
  static SbThreadMutex s_initThreadMutex;

  // The refs counter
  static int s_initRefCount;
};

#ifdef _WIN32
#pragma warning( pop )
#endif

#endif // SOCUDADEVICE_H

---

Open Inventor Toolkit reference manual, generated on 12 Feb 2024
Copyright © Thermo Fisher Scientific All rights reserved. http://www.openinventor.com/