boinc/lib/coproc.h

140 lines
3.7 KiB
C++

// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC is free software; you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC. If not, see <http://www.gnu.org/licenses/>.
#ifndef _COPROC_
#define _COPROC_
#include <vector>
#include <string>
#include <cstring>
#ifdef _USING_FCGI_
#include "fcgi_stdio.h"
#endif
#include "miofile.h"
#define MAX_COPROC_INSTANCES 8
struct COPROC {
char type[256]; // must be unique
int count; // how many are present
int used; // how many are in use (used by client)
void* owner[MAX_COPROC_INSTANCES];
// which ACTIVE_TASK each one is allocated to
#ifndef _USING_FCGI_
virtual void write_xml(MIOFILE&);
#endif
COPROC(const char* t){
strcpy(type, t);
count = 0;
used = 0;
memset(&owner, 0, sizeof(owner));
}
virtual void description(char*){};
virtual ~COPROC(){}
int parse(MIOFILE&);
};
struct COPROCS {
std::vector<COPROC*> coprocs; // not deleted in destructor
// so any structure that includes this needs to do it manually
COPROCS(){}
~COPROCS(){}
void delete_coprocs(){
for (unsigned int i=0; i<coprocs.size(); i++) {
delete coprocs[i];
}
}
#ifndef _USING_FCGI_
void write_xml(MIOFILE& out) {
for (unsigned int i=0; i<coprocs.size(); i++) {
coprocs[i]->write_xml(out);
}
}
#endif
std::vector<std::string> get();
int parse(FILE*);
COPROC* lookup(char*);
bool sufficient_coprocs(COPROCS&, bool verbose, const char* prefix);
void reserve_coprocs(COPROCS&, void*, bool verbose, const char* prefix);
void free_coprocs(COPROCS&, void*, bool verbose);
// Copy a coproc set, setting usage to zero.
// used in round-robin simulator and CPU scheduler,
// to avoid messing w/ master copy
//
void clone(COPROCS& c) {
for (unsigned int i=0; i<c.coprocs.size(); i++) {
COPROC* cp = c.coprocs[i];
COPROC* cp2 = new COPROC(cp->type);
cp2->count = cp->count;
coprocs.push_back(cp2);
}
}
};
// the following copied from /usr/local/cuda/include/driver_types.h
//
struct cudaDeviceProp {
char name[256];
size_t totalGlobalMem;
size_t sharedMemPerBlock;
int regsPerBlock;
int warpSize;
size_t memPitch;
int maxThreadsPerBlock;
int maxThreadsDim[3];
int maxGridSize[3];
int clockRate;
size_t totalConstMem;
int major;
int minor;
size_t textureAlignment;
int deviceOverlap;
int multiProcessorCount;
int __cudaReserved[40];
};
struct COPROC_CUDA : public COPROC {
cudaDeviceProp prop;
#ifndef _USING_FCGI_
virtual void write_xml(MIOFILE&);
#endif
COPROC_CUDA(): COPROC("CUDA"){}
virtual ~COPROC_CUDA(){}
static const char* get(COPROCS&);
virtual void description(char*);
void clear();
int parse(FILE*);
};
struct COPROC_CELL_SPE : public COPROC {
static const char* get(COPROCS&);
COPROC_CELL_SPE() : COPROC("Cell SPE"){}
virtual void description(char*);
virtual ~COPROC_CELL_SPE(){}
};
void fake_cuda(COPROCS&, int);
#endif