コード例 #1
0
ファイル: zpotf2_U.c プロジェクト: 34985086/meshlab
blasint CNAME(blas_arg_t *args, BLASLONG *range_m, BLASLONG *range_n, FLOAT *sa, FLOAT *sb, BLASLONG myid) {

  BLASLONG n, lda;
  FLOAT *a;

  FLOAT ajj[2];
  BLASLONG i, j;

  n      = args -> n;
  a      = (FLOAT *)args -> a;
  lda    = args -> lda;
  
  if (range_n) {
    n      = range_n[1] - range_n[0];
    a     += range_n[0] * (lda + 1) * COMPSIZE;
  }

  for (j = 0; j < n; j++) {

    ajj[0] = DOTC_K(j, a, 1, a, 1);
    GET_IMAGE(ajj[1]);

    ajj[0] = *(a + j * 2) - ajj[0]; 

    if (ajj[0] <= 0){
      *(a + j * 2 + 0) = ajj[0];
      *(a + j * 2 + 1) = ZERO;
      return j + 1;
    }

    ajj[0] = SQRT(ajj[0]);
    *(a + j * 2 + 0)  = ajj[0];
    *(a + j * 2 + 1)  = ZERO;

    i = n - j - 1;

    if (i > 0){
      GEMV_U(j, i, 0, dm1, ZERO,
	      a + lda * 2, lda,
	      a, 1,
	      a + (j + lda) * 2, lda, sb);
      
      SCAL_K(i, 0, 0, ONE / ajj[0], ZERO,
	      a + (j + lda) * 2, lda, NULL, 0, NULL, 0);
    }

    a += 2 * lda;
  }

  return 0;
}
コード例 #2
0
ファイル: zlauu2_L.c プロジェクト: AmEv7Fam/opentoonz
blasint CNAME(blas_arg_t *args, BLASLONG *range_m, BLASLONG *range_n, FLOAT *sa, FLOAT *sb, BLASLONG myid) {

  BLASLONG n, lda;
  FLOAT *a;

  FLOAT temp[2];
  BLASLONG i;

  n      = args -> n;
  a      = (FLOAT *)args -> a;
  lda    = args -> lda;
  
  if (range_n) {
    n      = range_n[1] - range_n[0];
    a     += range_n[0] * (lda + 1) * COMPSIZE;
  }

  for (i = 0; i < n; i++) {

    SCAL_K(i + 1, 0, 0, *(a + (i + i * lda) * COMPSIZE + 0), ZERO,
	    a + i * COMPSIZE, lda, NULL, 0, NULL, 0);
    
    if (i < n - 1) {
      temp[0] = DOTC_K(n - i - 1,
			a + (i + 1 + i * lda) * COMPSIZE, 1,
			a + (i + 1 + i * lda) * COMPSIZE, 1);
      GET_IMAGE(temp[1]);
      
      *(a + (i + i * lda) * COMPSIZE + 0) += temp[0];
      *(a + (i + i * lda) * COMPSIZE + 1)  = ZERO;
      
      GEMV_U(n - i - 1, i, 0, dp1, ZERO,
	     a + ((i + 1)          ) * COMPSIZE, lda,
	     a + ((i + 1) + i * lda) * COMPSIZE, 1,
	     a + ( i               ) * COMPSIZE , lda, sb);
    }
  }

  return 0;
}