SPU Samples

#include <stdio.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <sys/spu.h>
#include "spu_bin.h"

int main(int argc, char *argv[])
{
    u32 spu_id = 0;
    sysSpuImage image;

    printf("sputest starting....\n");

    // Initialize SPU runtime (6 SPUs, 5 raw SPUs)
    printf("Initializing 6 SPUs...\n");
    sysSpuInitialize(6, 5);

    // Create a raw SPU thread
    printf("Initializing raw SPU...\n");
    sysSpuRawCreate(&spu_id, NULL);

    // Import SPU binary image
    printf("Importing spu image...\n");
    sysSpuImageImport(&image, spu_bin, SPU_IMAGE_PROTECT);

    // Load image into SPU local store
    printf("Loading spu image into SPU %d...\n", spu_id);
    sysSpuRawImageLoad(spu_id, &image);

    // Start SPU execution
    printf("Starting SPU %d...\n", spu_id);
    sysSpuRawWriteProblemStorage(spu_id, SPU_RunCtrl, 1);

    // Wait for SPU to write to outbound mailbox
    printf("Waiting for SPU to return...\n");
    while (!(sysSpuRawReadProblemStorage(spu_id, SPU_MBox_Status) & 1));

    // Read mailbox value
    printf("SPU Mailbox return value: %08x\n",
           sysSpuRawReadProblemStorage(spu_id, SPU_Out_MBox));

    // Cleanup
    printf("Destroying SPU %d...\n", spu_id);
    sysSpuRawDestroy(spu_id);

    printf("Closing SPU image...\n");
    sysSpuImageClose(&image);

    return 0;
}

#include <spu_intrinsics.h>

int main()
{
    // Write a value to outbound mailbox
    spu_writech(SPU_WrOutMbox, 0x1337BAAD);
    return 0;
}

#include <stdio.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <malloc.h>
#include <sys/spu.h>
#include "spustr.h"
#include "spu_bin.h"

#define ptr2ea(x) ((u64)((void*)(x)))

int main(int argc, char *argv[])
{
    u32 *array;
    u32 group_id;
    spustr_t *spu;
    sysSpuImage image;
    u32 cause, status, i;
    sysSpuThreadArgument arg[6];
    sysSpuThreadGroupAttribute grpattr = {
        7+1, ptr2ea("mygroup"), 0, 0
    };
    sysSpuThreadAttribute attr = {
        ptr2ea("mythread"), 8+1, SPU_THREAD_ATTR_NONE
    };

    printf("spuchain starting....\n");

    // Initialize SPU subsystem
    sysSpuInitialize(6, 0);
    
    // Load SPU program
    sysSpuImageImport(&image, spu_bin, 0);
    
    // Create thread group
    sysSpuThreadGroupCreate(&group_id, 6, 100, &grpattr);

    // Allocate shared data
    spu = (spustr_t*)memalign(128, 6*sizeof(spustr_t));
    array = (u32*)memalign(128, 4*sizeof(u32));

    // Initialize and create 6 SPU threads
    for(i = 0; i < 6; i++) {
        spu[i].rank = i;
        spu[i].count = 6;
        spu[i].sync = 0;
        spu[i].array_ea = ptr2ea(array);
        arg[i].arg0 = ptr2ea(&spu[i]);

        printf("Creating SPU thread... ");
        sysSpuThreadInitialize(&spu[i].id, group_id, i,
                               &image, &attr, &arg[i]);
        printf("%08x\n", spu[i].id);
        
        // Configure signal notification mode
        sysSpuThreadSetConfiguration(spu[i].id,
            (SPU_SIGNAL1_OVERWRITE | SPU_SIGNAL2_OVERWRITE));
    }

    // Start all SPU threads
    printf("Starting SPU thread group....\n");
    sysSpuThreadGroupStart(group_id);

    // Initialize array
    printf("Initial array:");
    for(i = 0; i < 4; i++) {
        array[i] = (i + 1);
        printf(" %d", array[i]);
    }
    printf("\n");

    // Trigger the chain by signaling SPU 0
    printf("sending signal.... \n");
    sysSpuThreadWriteSignal(spu[0].id, 0, 1);

    // Wait for SPU 5 to complete
    while(spu[5].sync == 0);

    // Display results
    printf("Output array:");
    for(i = 0; i < 4; i++)
        printf(" %d", array[i]);
    printf("\n");

    // Cleanup
    printf("Joining SPU thread group....\n");
    sysSpuThreadGroupJoin(group_id, &cause, &status);
    sysSpuImageClose(&image);

    free(array);
    free(spu);

    return 0;
}