splitstruct.br

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

typedef struct NHType_t
{
    float3 row1;
    float3 row2;
    float3 row3;// 3X3 matrix
} NHType;


kernel void Cinv_kernel(float2 constants, NHType F<>, 
			out NHType Cinv<>,
			out float q<>)
{

   Cinv.row1 = float3(1,2,3);
   Cinv.row2 = float3(2,3,4);
   Cinv.row3 = float3(5,6,7);
   
   q = 8;
}


kernel void Cinv_kernel_nostruct(float2 constants, NHType F<>, 
				 out float3 Cinv1<>, 
                                 out float3 Cinv2<>, 
                                 out float3 Cinv3<>,
				 out float q<>)
{

   Cinv1 = float3(1,2,3);
   Cinv2 = float3(2,3,4);
   Cinv3 = float3(5,6,7);

   q = 8;
}

#define NUM 100

NHType	  S[NUM];
float3	  S1[NUM];
float3	  S2[NUM];
float3	  S3[NUM];
NHType    Cinvstream<NUM>;
float3    Cinvstream1<NUM>;
float3    Cinvstream2<NUM>;
float3    Cinvstream3<NUM>;
NHType    F<NUM>;
float     q[NUM];
float     r[NUM];
float     qstream<NUM>;
float     q2stream<NUM>;

int main (void) {

  streamRead(Cinvstream, S);
  streamRead(Cinvstream1, S1);
  streamRead(Cinvstream2, S2);
  streamRead(Cinvstream3, S3);
  streamRead(qstream,  q);
  streamRead(q2stream, q);

  Cinv_kernel (float2(3.0f,2.0f), F, Cinvstream, qstream);

  streamWrite(Cinvstream, S);
  streamWrite(qstream, q);

  printf ("%3.2f %3.2f %3.2f\n", S[0].row1.x, S[0].row1.y, S[0].row1.z);
  printf ("%3.2f %3.2f %3.2f\n", S[0].row2.x, S[0].row2.y, S[0].row2.z);
  printf ("%3.2f %3.2f %3.2f\n", S[0].row3.x, S[0].row3.y, S[0].row3.z);
  printf ("%3.2f\n\n", q[0]);

  // Same thing except we eliminated the structure
  Cinv_kernel_nostruct (float2(3.0f,2.0f), F, 
			Cinvstream1, Cinvstream2, Cinvstream3,
			q2stream);

  streamWrite(Cinvstream1, S1);
  streamWrite(Cinvstream2, S2);
  streamWrite(Cinvstream3, S3);
  streamWrite(q2stream, q);

  printf ("%3.2f %3.2f %3.2f\n", S1[0].x, S1[0].y, S1[0].z);
  printf ("%3.2f %3.2f %3.2f\n", S2[0].x, S2[0].y, S2[0].z);
  printf ("%3.2f %3.2f %3.2f\n", S3[0].x, S3[0].y, S3[0].z);
  printf ("%3.2f\n\n", q[0]);
  return 0;	
}