domain_write.br

来自「用于GPU通用计算的编程语言BrookGPU 0.4」· BR 代码 · 共 61 行

// domain_write.br
// tests that writing to a domain works

#include <stdio.h>

kernel void copy( float input<>, out float output<> )
{
  output = input;
}

int main()
{
	int i, j;
	float a1< 100 >;
	float b1< 100 >;
	float a2< 10, 10 >;
	float b2< 10, 10 >;

	float a1_data[100];
	float b1_data[100];
	float a2_data[100];
	float b2_data[100];
	
	// initialize data
	for( i = 0; i < 100; i++ )
	{
		a1_data[i] = (float)i;
		a2_data[i] = (float)i;
		b1_data[i] = -1.0f;
		b2_data[i] = -1.0f;
	}
	streamRead( a1, a1_data );
	streamRead( b1, b1_data );
	streamRead( a2, a2_data );
	streamRead( b2, b2_data );
	
	copy( a1.domain( 30, 80 ), b1.domain( 20, 70 ) );
	copy( a2.domain( int2(3,5), int2(9,10) ), b2.domain( int2(0,2), int2(6,7) ) );
	
	streamWrite( b1, b1_data );
	streamWrite( b2, b2_data );
	
	for( i = 0; i < 10; i++ )
	{
		for( j = 0; j < 10; j++ )
			printf( "%3.2f ", b1_data[10*i+j] );
		printf("\n");
	}
	
	printf("***\n");
	
	for( i = 0; i < 10; i++ )
	{
		for( j = 0; j < 10; j++ )
			printf( "%3.2f ", b2_data[10*i+j] );
		printf("\n");
	}
	
	return 0;
}