sum.br

用于GPU通用计算的编程语言BrookGPU 0.4

// sum.br
// The absolute minimal test. It tests
// using a kernel to sum both 1-dimensional
// and two-dimensional input streams.

#include <stdio.h>

kernel void sum( float a<>, float b<>, out float c<> ) {
	c = a + b;
}

int main() {
	int i, j;
	float a1< 100 >;
	float b1< 100 >;
	float c1< 100 >;
	float a2< 10, 10 >;
	float b2< 10, 10 >;
	float c2< 10, 10 >;
	float input_a1[100];
	float input_b1[100];
	float output_c1[100];
	float input_a2[10][10];
	float input_b2[10][10];
	float output_c2[10][10];
	
	// initialize data
	for( i = 0; i < 10; i++ )
	{
		for( j = 0; j < 10; j++ )
		{
			input_a1[10*i+j] = (float)i;
			input_b1[10*i+j] = (float)10*j;
			input_a2[i][j] = (float)i;
			input_b2[i][j] = (float)10*j;
		}
	}
	streamRead( a1, input_a1 );
	streamRead( b1, input_b1 );
	streamRead( a2, input_a1 );
	streamRead( b2, input_b2 );
	
	sum( a1, b1, c1 );
	sum( a2, b2, c2 );
	
	streamWrite( c1, output_c1 );
	streamWrite( c2, output_c2 );
	
	for( i = 0; i < 10; i++ )
	{
		for( j = 0; j < 10; j++ )
			printf( "%3.2f ", output_c1[10*i+j] );
		printf("\n");
	}
	
	for( i = 0; i < 10; i++ )
	{
		for( j = 0; j < 10; j++ )
			printf( "%3.2f ", output_c2[i][j] );
		printf("\n");
	}
	
	return 0;
}