big.br

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

// big.br
// tests big streams with address-translation

#include <stdio.h>

kernel void copy( float a[][][][], out float b<> ) {
        float4 index = {(indexof b).x,0,0,0};
	b = a[index];
}

int main( int argc, char** argv ) {
	int SIZE = atoi( argv[1] );
	int i, j;
	
	float a< 1, 1, 1, SIZE >;
	float b< SIZE >;
	float* a_data;
	float* b_data;
	
	a_data = (float*)malloc( SIZE*sizeof(float) );
	b_data = (float*)malloc( SIZE*sizeof(float) );

	for( i = 0; i < SIZE; i++ )
	{
		a_data[i] = (float)i;
	}

	streamRead( a, a_data );
	copy( a, b );
	streamWrite( b, b_data );

	for( i = 0; i < SIZE; i++ )
	{
		j = i;
		if( b_data[i] != j )
			printf( "[%d] %f != %d\n", i, b_data[i], j );
	}
	printf( "done\n" );

	return 0;
}