int KernelProbe::outputState(double timed) {
InterColComm * icComm = parent->icCommunicator();
const int rank = icComm->commRank();
if( rank != 0 ) return PV_SUCCESS;
assert(getTargetConn()!=NULL);
int nxp = getTargetHyPerConn()->xPatchSize();
int nyp = getTargetHyPerConn()->yPatchSize();
int nfp = getTargetHyPerConn()->fPatchSize();
int patchSize = nxp*nyp*nfp;
const pvwdata_t * wdata = getTargetHyPerConn()->get_wDataStart(arborID)+patchSize*kernelIndex;
const pvwdata_t * dwdata = outputPlasticIncr ?
getTargetHyPerConn()->get_dwDataStart(arborID)+patchSize*kernelIndex : NULL;
fprintf(outputstream->fp, "Time %f, Conn \"%s\", nxp=%d, nyp=%d, nfp=%d\n",
timed, getTargetConn()->getName(),nxp, nyp, nfp);
for(int f=0; f<nfp; f++) {
for(int y=0; y<nyp; y++) {
for(int x=0; x<nxp; x++) {
int k = kIndex(x,y,f,nxp,nyp,nfp);
fprintf(outputstream->fp, " x=%d, y=%d, f=%d (index %d):", x, y, f, k);
if(getOutputWeights()) {
fprintf(outputstream->fp, " weight=%f", (float)wdata[k]);
}
if(getOutputPlasticIncr()) {
fprintf(outputstream->fp, " dw=%f", (float)dwdata[k]);
}
fprintf(outputstream->fp,"\n");
}
}
}
return PV_SUCCESS;
}
int TestConnProbe::outputState(double timed){
//Grab weights of probe and test for the value of .625/1.5, or .4166666
HyPerConn* conn = getTargetHyPerConn();
int numPreExt = conn->preSynapticLayer()->getNumExtended();
int syw = conn->yPatchStride(); // stride in patch
for(int kPre = 0; kPre < numPreExt; kPre++){
PVPatch * weights = conn->getWeights(kPre, 0);
int nk = conn->fPatchSize() * weights->nx;
pvwdata_t * data = conn->get_wData(0,kPre);
int ny = weights->ny;
for (int y = 0; y < ny; y++) {
pvwdata_t * dataYStart = data + y * syw;
for(int k = 0; k < nk; k++){
if(fabs(timed - 0) < (parent->getDeltaTime()/2)){
if(fabs(dataYStart[k] - 1) > .01){
std::cout << "dataYStart[k]: " << dataYStart[k] << "\n";
}
assert(fabs(dataYStart[k] - 1) <= .01);
}
else if(fabs(timed - 1) < (parent->getDeltaTime()/2)){
if(fabs(dataYStart[k] - 1.375) > .01){
std::cout << "dataYStart[k]: " << dataYStart[k] << "\n";
}
assert(fabs(dataYStart[k] - 1.375) <= .01);
}
}
}
}
return PV_SUCCESS;
}
int MomentumTestConnProbe::outputState(double timed){
//Grab weights of probe and test for the value of .625/1.5, or .4166666
HyPerConn* conn = getTargetHyPerConn();
int numPreExt = conn->preSynapticLayer()->getNumExtended();
int syw = conn->yPatchStride(); // stride in patch
for(int kPre = 0; kPre < numPreExt; kPre++){
PVPatch * weights = conn->getWeights(kPre, 0);
int nk = conn->fPatchSize() * weights->nx;
pvwdata_t * data = conn->get_wData(0,kPre);
int ny = weights->ny;
pvdata_t wCorrect;
for (int y = 0; y < ny; y++) {
pvwdata_t * dataYStart = data + y * syw;
for(int k = 0; k < nk; k++){
pvdata_t wObserved = dataYStart[k];
if(timed < 3){
wCorrect = 0;
}
else{
wCorrect = .376471;
for(int i = 0; i < (timed-3); i++){
wCorrect += .376471 * exp(-(2*(i+1)));
}
}
assert(fabs(wObserved - wCorrect) <= 1e-4);
}
}
}
return PV_SUCCESS;
}
int KernelProbe::allocateDataStructures() {
int status = PV_SUCCESS;
assert(getTargetConn());
if (getKernelIndex()<0 || getKernelIndex()>=getTargetHyPerConn()->getNumDataPatches()) {
fprintf(stderr, "KernelProbe \"%s\": kernelIndex %d is out of bounds. (min 0, max %d)\n", name, getKernelIndex(), getTargetHyPerConn()->getNumDataPatches()-1);
exit(EXIT_FAILURE);
}
if (getArbor()<0 || getArbor()>=getTargetConn()->numberOfAxonalArborLists()) {
fprintf(stderr, "KernelProbe \"%s\": arborId %d is out of bounds. (min 0, max %d)\n", name, getArbor(), getTargetConn()->numberOfAxonalArborLists()-1);
exit(EXIT_FAILURE);
}
if(outputstream) {
fprintf(outputstream->fp, "Probe \"%s\", kernel index %d, arbor index %d.\n", name, getKernelIndex(), getArbor());
}
if(getOutputPatchIndices()) {
patchIndices(getTargetHyPerConn());
}
return status;
}
/**
* @timef
*/
int PlasticConnTestProbe::outputState(double timed) {
HyPerConn * c = getTargetHyPerConn();
InterColComm * icComm = c->getParent()->icCommunicator();
const int rcvProc = 0;
if( icComm->commRank() != rcvProc ) {
return PV_SUCCESS;
}
assert(getTargetConn()!=NULL);
outputStream->printf(" Time %f, connection \"%s\":\n", timed, getTargetName());
const pvwdata_t * w = c->get_wDataHead(getArbor(), getKernelIndex());
const pvdata_t * dw = c->get_dwDataHead(getArbor(), getKernelIndex());
if( getOutputPlasticIncr() && dw == NULL ) {
pvError().printf("PlasticConnTestProbe \"%s\": connection \"%s\" has dKernelData(%d,%d) set to null.\n", getName(), getTargetName(), getKernelIndex(), getArbor());
}
int nxp = c->xPatchSize();
int nyp = c->yPatchSize();
int nfp = c->fPatchSize();
int status = PV_SUCCESS;
for( int k=0; k<nxp*nyp*nfp; k++ ) {
int x=kxPos(k,nxp,nyp,nfp);
int wx = (nxp-1)/2 - x; // assumes connection is one-to-one
if(getOutputWeights()) {
pvdata_t wCorrect = timed*wx;
pvdata_t wObserved = w[k];
if( fabs( ((double) (wObserved - wCorrect))/timed ) > 1e-4 ) {
int y=kyPos(k,nxp,nyp,nfp);
int f=featureIndex(k,nxp,nyp,nfp);
outputStream->printf(" index %d (x=%d, y=%d, f=%d: w = %f, should be %f\n", k, x, y, f, wObserved, wCorrect);
}
}
if(timed > 0 && getOutputPlasticIncr() && dw != NULL) {
pvdata_t dwCorrect = wx;
pvdata_t dwObserved = dw[k];
if( dwObserved != dwCorrect ) {
int y=kyPos(k,nxp,nyp,nfp);
int f=featureIndex(k,nxp,nyp,nfp);
outputStream->printf(" index %d (x=%d, y=%d, f=%d: dw = %f, should be %f\n", k, x, y, f, dwObserved, dwCorrect);
}
}
}
assert(status==PV_SUCCESS);
if( status == PV_SUCCESS ) {
if (getOutputWeights()) { outputStream->printf(" All weights are correct.\n"); }
if (getOutputPlasticIncr()) { outputStream->printf(" All plastic increments are correct.\n"); }
}
if(getOutputPatchIndices()) {
patchIndices(c);
}
return PV_SUCCESS;
}
/**
* @timef
*/
int MomentumConnTestProbe::outputState(double timed) {
HyPerConn * c = getTargetHyPerConn();
InterColComm * icComm = c->getParent()->icCommunicator();
const int rcvProc = 0;
if( icComm->commRank() != rcvProc ) {
return PV_SUCCESS;
}
assert(getTargetConn()!=NULL);
FILE * fp = getStream()->fp;
fprintf(fp, " Time %f, connection \"%s\":\n", timed, getTargetName());
const pvwdata_t * w = c->get_wDataHead(getArbor(), getKernelIndex());
const pvdata_t * dw = c->get_dwDataHead(getArbor(), getKernelIndex());
if( getOutputPlasticIncr() && dw == NULL ) {
fprintf(stderr, "MomentumConnTestProbe \"%s\": connection \"%s\" has dKernelData(%d,%d) set to null.\n", getName(), getTargetName(), getKernelIndex(), getArbor());
assert(false);
}
int nxp = c->xPatchSize();
int nyp = c->yPatchSize();
int nfp = c->fPatchSize();
int status = PV_SUCCESS;
for( int k=0; k<nxp*nyp*nfp; k++ ) {
pvdata_t wObserved = w[k];
//Pulse happens at time 3
pvdata_t wCorrect;
if(timed < 3){
wCorrect = 0;
}
else{
if(isViscosity){
wCorrect = 1;
for(int i = 0; i < (timed - 3); i++){
wCorrect += exp(-(2*(i+1)));
}
}
else{
wCorrect = 2 - pow(2, -(timed - 3));
}
}
if( fabs( ((double) (wObserved - wCorrect))/timed ) > 1e-4 ) {
int y=kyPos(k,nxp,nyp,nfp);
int f=featureIndex(k,nxp,nyp,nfp);
fprintf(fp, " w = %f, should be %f\n", wObserved, wCorrect);
exit(-1);
}
}
return PV_SUCCESS;
}
int KernelProbe::communicateInitInfo() {
int status = BaseHyPerConnProbe::communicateInitInfo();
assert(targetHyPerConn);
if(getTargetHyPerConn()->usingSharedWeights()==false) {
if (parent->columnId()==0) {
fprintf(stderr, "KernelProbe \"%s\": connection \"%s\" is not using shared weights.\n", name, targetConn->getName());
}
status = PV_FAILURE;
}
MPI_Barrier(parent->icCommunicator()->communicator());
if (status != PV_SUCCESS) {
exit(EXIT_FAILURE);
}
return status;
}
/**
* @timef
* NOTES:
* - kPost, kxPost, kyPost are indices in the restricted post-synaptic layer.
*
*/
int PostConnProbe::outputState(double timef)
{
int k, kxPre, kyPre;
HyPerConn * c = getTargetHyPerConn();
PVPatch * w;
PVPatch *** wPost = c->convertPreSynapticWeights(timef);
// TODO - WARNING: currently only works if nfPre==0
const PVLayer * lPre = c->preSynapticLayer()->clayer;
const PVLayer * lPost = c->postSynapticLayer()->clayer;
const int nxPre = lPre->loc.nx;
const int nyPre = lPre->loc.ny;
const int nfPre = lPre->loc.nf;
const PVHalo * haloPre = &lPre->loc.halo;
const int nxPost = lPost->loc.nx;
const int nyPost = lPost->loc.ny;
const int nfPost = lPost->loc.nf;
const PVHalo * haloPost = &lPost->loc.halo;
// calc kPost if needed
if (kPost < 0) {
kPost = kIndex(kxPost, kyPost, kfPost, nxPost, nyPost, nfPost);
}
else {
kxPost = kxPos(kPost, nxPost, nyPost, nfPost);
kyPost = kyPos(kPost, nxPost, nyPost, nfPost);
kfPost = featureIndex(kPost, nxPost, nyPost, nfPost);
}
c->preSynapticPatchHead(kxPost, kyPost, kfPost, &kxPre, &kyPre);
const int kxPreEx = kxPre + haloPre->lt;
const int kyPreEx = kyPre + haloPre->up;
const int kxPostEx = kxPost + haloPost->lt;
const int kyPostEx = kyPost + haloPost->up;
const int kPostEx = kIndex(kxPostEx, kyPostEx, kfPost, nxPost+haloPost->lt+haloPost->rt, nyPost+haloPost->dn+haloPost->up, nfPost);
const bool postFired = lPost->activity->data[kPostEx] > 0.0;
w = wPost[getArborID()][kPost];
pvwdata_t * wPostData = c->getWPostData(getArborID(),kPost);
const int nw = w->nx * w->ny * nfPost; //w->nf;
if (wPrev == NULL) {
wPrev = (pvwdata_t *) calloc(nw, sizeof(pvwdata_t));
for (k = 0; k < nw; k++) {
wPrev[k] = wPostData[k]; // This is broken if the patch is shrunken
}
}
if (wActiv == NULL) {
wActiv = (pvwdata_t *) calloc(nw, sizeof(pvwdata_t));
}
k = 0;
for (int ky = 0; ky < w->ny; ky++) {
for (int kx = 0; kx < w->nx; kx++) {
int kPre = kIndex(kx+kxPreEx, ky+kyPreEx, 0, nxPre+haloPre->lt+haloPre->rt, nyPre+haloPre->dn+haloPre->up, nfPre);
wActiv[k++] = lPre->activity->data[kPre];
}
}
bool changed = false;
for (k = 0; k < nw; k++) {
if (wPrev[k] != wPostData[k] || wActiv[k] != 0.0) {
changed = true;
break;
}
}
FILE * fp = getStream()->fp;
if (stdpVars && (postFired || changed)) {
if (postFired) fprintf(fp, "*");
else fprintf(fp, " ");
fprintf(fp, "t=%.1f w%d(%d,%d,%d) prePatchHead(%d,%d): ", timef, kPost, kxPost,
kyPost, kfPost, kxPre, kyPre);
if (image) fprintf(fp, "tag==%d ", image->tag());
fprintf(fp, "\n");
}
if (stdpVars && changed) {
text_write_patch_extra(fp, w, wPostData, wPrev, wActiv, getTargetHyPerConn());
fflush(fp);
}
for (k = 0; k < nw; k++) {
wPrev[k] = wPostData[k];
}
if (outputIndices) {
fprintf(fp, "w%d(%d,%d,%d) prePatchHead(%d,%d): ", kPost, kxPost, kyPost, kfPost, kxPre, kyPre);
if(!stdpVars){
fprintf(fp,"\n");
}
const PVLayer * lPre = c->preSynapticLayer()->clayer;
write_patch_indices(fp, w, &lPre->loc, kxPre, kyPre, 0);
fflush(fp);
}
return 0;
}
