mirror of
https://github.com/CNugteren/CLBlast.git
synced 2024-08-26 23:07:00 +02:00
105 lines
4.2 KiB
C++
105 lines
4.2 KiB
C++
|
|
// =================================================================================================
|
|
// This file is part of the CLBlast project. The project is licensed under Apache Version 2.0. This
|
|
// project loosely follows the Google C++ styleguide and uses a tab-size of two spaces and a max-
|
|
// width of 100 characters per line.
|
|
//
|
|
// Author(s):
|
|
// Cedric Nugteren <www.cedricnugteren.nl>
|
|
//
|
|
// This file implements the Database class, providing a static variable holding the actual database
|
|
// information. The class also provides utility functions to search the database and to access a
|
|
// found entry by parameter-key. The database itself is filled in the corresponding source-file and
|
|
// partially also by the database/xxxxx.h files, in which kernel-specific parameters are found.
|
|
//
|
|
// =================================================================================================
|
|
|
|
#ifndef CLBLAST_DATABASE_H_
|
|
#define CLBLAST_DATABASE_H_
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <unordered_map>
|
|
|
|
#include "internal/utilities.h"
|
|
|
|
namespace clblast {
|
|
// =================================================================================================
|
|
|
|
// See comment at top of file for a description of the class
|
|
class Database {
|
|
public:
|
|
|
|
// Type alias for the database parameters
|
|
using Parameters = std::unordered_map<std::string,size_t>;
|
|
|
|
// Structures for content inside the database
|
|
struct DatabaseDevice {
|
|
const std::string name;
|
|
const Parameters parameters;
|
|
};
|
|
struct DatabaseVendor {
|
|
const std::string type;
|
|
const std::string name;
|
|
const std::vector<DatabaseDevice> devices;
|
|
};
|
|
struct DatabaseEntry {
|
|
const std::string kernel;
|
|
const Precision precision;
|
|
const std::vector<DatabaseVendor> vendors;
|
|
};
|
|
|
|
// The OpenCL device types
|
|
static constexpr auto kDeviceTypeCPU = "CPU";
|
|
static constexpr auto kDeviceTypeGPU = "GPU";
|
|
static constexpr auto kDeviceTypeAccelerator = "accelerator";
|
|
static constexpr auto kDeviceTypeAll = "default";
|
|
|
|
// The OpenCL device vendors
|
|
static constexpr auto kDeviceVendorAll = "default";
|
|
|
|
// Alternative names for some OpenCL vendors
|
|
const std::unordered_map<std::string,std::string> kVendorNames {
|
|
{"Intel(R) Corporation", "Intel"},
|
|
{"GenuineIntel", "Intel"},
|
|
{"Advanced Micro Devices, Inc.", "AMD"},
|
|
{"NVIDIA Corporation", "NVIDIA"},
|
|
};
|
|
|
|
// The database consists of separate database entries, stored together in a vector
|
|
static const DatabaseEntry XaxpySingle, XaxpyDouble, XaxpyComplexSingle, XaxpyComplexDouble;
|
|
static const DatabaseEntry XdotSingle, XdotDouble, XdotComplexSingle, XdotComplexDouble;
|
|
static const DatabaseEntry XgemvSingle, XgemvDouble, XgemvComplexSingle, XgemvComplexDouble;
|
|
static const DatabaseEntry XgerSingle, XgerDouble, XgerComplexSingle, XgerComplexDouble;
|
|
static const DatabaseEntry XgemmSingle, XgemmDouble, XgemmComplexSingle, XgemmComplexDouble;
|
|
static const DatabaseEntry CopySingle, CopyDouble, CopyComplexSingle, CopyComplexDouble;
|
|
static const DatabaseEntry PadSingle, PadDouble, PadComplexSingle, PadComplexDouble;
|
|
static const DatabaseEntry TransposeSingle, TransposeDouble, TransposeComplexSingle, TransposeComplexDouble;
|
|
static const DatabaseEntry PadtransposeSingle, PadtransposeDouble, PadtransposeComplexSingle, PadtransposeComplexDouble;
|
|
static const std::vector<DatabaseEntry> database;
|
|
|
|
// The constructor
|
|
explicit Database(const Queue &queue, const std::vector<std::string> &routines,
|
|
const Precision precision);
|
|
|
|
// Accessor of values by key
|
|
size_t operator[](const std::string key) const { return parameters_.find(key)->second; }
|
|
|
|
// Obtain a list of OpenCL pre-processor defines based on the parameters
|
|
std::string GetDefines() const;
|
|
|
|
private:
|
|
Parameters Search(const std::string &this_kernel, const std::string &this_type,
|
|
const std::string &this_vendor, const std::string &this_device,
|
|
const Precision this_precision) const;
|
|
|
|
// Found parameters suitable for this device/kernel
|
|
Parameters parameters_;
|
|
};
|
|
|
|
// =================================================================================================
|
|
} // namespace clblast
|
|
|
|
// CLBLAST_DATABASE_H_
|
|
#endif
|