input_output.br

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

// input_output.br
// Trivial regression test to confirm that
// a runtime correctly handles using the
// same stream as both an input and output
// argument to a kernel.

#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 a2< 10, 10 >;
	float b2< 10, 10 >;
	float input_a1[100];
	float input_b1[100];
	float output_b1[100];
	float input_a2[10][10];
	float input_b2[10][10];
	float output_b2[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, b1 );
	sum( a2, b2, b2 );
	
	streamWrite( b1, output_b1 );
	streamWrite( b2, output_b2 );
	
	for( i = 0; i < 10; i++ )
	{
		for( j = 0; j < 10; j++ )
			printf( "%3.2f ", output_b1[10*i+j] );
		printf("\n");
	}
	
	for( i = 0; i < 10; i++ )
	{
		for( j = 0; j < 10; j++ )
			printf( "%3.2f ", output_b2[i][j] );
		printf("\n");
	}
	
	return 0;
}