cannam@95: /*
cannam@95:  * Copyright (c) 2001 Matteo Frigo
cannam@95:  * Copyright (c) 2001 Massachusetts Institute of Technology
cannam@95:  *
cannam@95:  * This program is free software; you can redistribute it and/or modify
cannam@95:  * it under the terms of the GNU General Public License as published by
cannam@95:  * the Free Software Foundation; either version 2 of the License, or
cannam@95:  * (at your option) any later version.
cannam@95:  *
cannam@95:  * This program is distributed in the hope that it will be useful,
cannam@95:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
cannam@95:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
cannam@95:  * GNU General Public License for more details.
cannam@95:  *
cannam@95:  * You should have received a copy of the GNU General Public License
cannam@95:  * along with this program; if not, write to the Free Software
cannam@95:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
cannam@95:  *
cannam@95:  */
cannam@95: 
cannam@95: #include "bench.h"
cannam@95: #include <stdlib.h>
cannam@95: 
cannam@95: bench_tensor *mktensor(int rnk) 
cannam@95: {
cannam@95:      bench_tensor *x;
cannam@95: 
cannam@95:      BENCH_ASSERT(rnk >= 0);
cannam@95: 
cannam@95:      x = (bench_tensor *)bench_malloc(sizeof(bench_tensor));
cannam@95:      if (FINITE_RNK(rnk) && rnk > 0)
cannam@95:           x->dims = (bench_iodim *)bench_malloc(sizeof(bench_iodim) * rnk);
cannam@95:      else
cannam@95:           x->dims = 0;
cannam@95: 
cannam@95:      x->rnk = rnk;
cannam@95:      return x;
cannam@95: }
cannam@95: 
cannam@95: void tensor_destroy(bench_tensor *sz)
cannam@95: {
cannam@95:      bench_free0(sz->dims);
cannam@95:      bench_free(sz);
cannam@95: }
cannam@95: 
cannam@95: int tensor_sz(const bench_tensor *sz)
cannam@95: {
cannam@95:      int i, n = 1;
cannam@95: 
cannam@95:      if (!FINITE_RNK(sz->rnk))
cannam@95:           return 0;
cannam@95: 
cannam@95:      for (i = 0; i < sz->rnk; ++i)
cannam@95:           n *= sz->dims[i].n;
cannam@95:      return n;
cannam@95: }
cannam@95: 
cannam@95: 
cannam@95: /* total order among bench_iodim's */
cannam@95: static int dimcmp(const bench_iodim *a, const bench_iodim *b)
cannam@95: {
cannam@95:      if (b->is != a->is)
cannam@95:           return (b->is - a->is);	/* shorter strides go later */
cannam@95:      if (b->os != a->os)
cannam@95:           return (b->os - a->os);	/* shorter strides go later */
cannam@95:      return (int)(a->n - b->n);	        /* larger n's go later */
cannam@95: }
cannam@95: 
cannam@95: bench_tensor *tensor_compress(const bench_tensor *sz)
cannam@95: {
cannam@95:      int i, rnk;
cannam@95:      bench_tensor *x;
cannam@95: 
cannam@95:      BENCH_ASSERT(FINITE_RNK(sz->rnk));
cannam@95:      for (i = rnk = 0; i < sz->rnk; ++i) {
cannam@95:           BENCH_ASSERT(sz->dims[i].n > 0);
cannam@95:           if (sz->dims[i].n != 1)
cannam@95:                ++rnk;
cannam@95:      }
cannam@95: 
cannam@95:      x = mktensor(rnk);
cannam@95:      for (i = rnk = 0; i < sz->rnk; ++i) {
cannam@95:           if (sz->dims[i].n != 1)
cannam@95:                x->dims[rnk++] = sz->dims[i];
cannam@95:      }
cannam@95: 
cannam@95:      if (rnk) {
cannam@95: 	  /* God knows how qsort() behaves if n==0 */
cannam@95: 	  qsort(x->dims, (size_t)x->rnk, sizeof(bench_iodim),
cannam@95: 		(int (*)(const void *, const void *))dimcmp);
cannam@95:      }
cannam@95: 
cannam@95:      return x;
cannam@95: }
cannam@95: 
cannam@95: int tensor_unitstridep(bench_tensor *t)
cannam@95: {
cannam@95:      BENCH_ASSERT(FINITE_RNK(t->rnk));
cannam@95:      return (t->rnk == 0 ||
cannam@95: 	     (t->dims[t->rnk - 1].is == 1 && t->dims[t->rnk - 1].os == 1));
cannam@95: }
cannam@95: 
cannam@95: /* detect screwy real padded rowmajor... ugh */
cannam@95: int tensor_real_rowmajorp(bench_tensor *t, int sign, int in_place)
cannam@95: {
cannam@95:      int i;
cannam@95: 
cannam@95:      BENCH_ASSERT(FINITE_RNK(t->rnk));
cannam@95: 
cannam@95:      i = t->rnk - 1;
cannam@95: 
cannam@95:      if (--i >= 0) {
cannam@95:           bench_iodim *d = t->dims + i;
cannam@95: 	  if (sign < 0) {
cannam@95: 	       if (d[0].is != d[1].is * (in_place ? 2*(d[1].n/2 + 1) : d[1].n))
cannam@95: 		    return 0;
cannam@95: 	       if (d[0].os != d[1].os * (d[1].n/2 + 1))
cannam@95: 		    return 0;
cannam@95: 	  }
cannam@95: 	  else {
cannam@95: 	       if (d[0].is != d[1].is * (d[1].n/2 + 1))
cannam@95: 		    return 0;
cannam@95: 	       if (d[0].os != d[1].os * (in_place ? 2*(d[1].n/2 + 1) : d[1].n))
cannam@95: 		    return 0;
cannam@95: 	  }
cannam@95:      }
cannam@95: 
cannam@95:      while (--i >= 0) {
cannam@95:           bench_iodim *d = t->dims + i;
cannam@95:           if (d[0].is != d[1].is * d[1].n)
cannam@95:                return 0;
cannam@95:           if (d[0].os != d[1].os * d[1].n)
cannam@95:                return 0;
cannam@95:      }
cannam@95:      return 1;
cannam@95: }
cannam@95: 
cannam@95: int tensor_rowmajorp(bench_tensor *t)
cannam@95: {
cannam@95:      int i;
cannam@95: 
cannam@95:      BENCH_ASSERT(FINITE_RNK(t->rnk));
cannam@95: 
cannam@95:      i = t->rnk - 1;
cannam@95:      while (--i >= 0) {
cannam@95: 	  bench_iodim *d = t->dims + i;
cannam@95: 	  if (d[0].is != d[1].is * d[1].n)
cannam@95: 	       return 0;
cannam@95: 	  if (d[0].os != d[1].os * d[1].n)
cannam@95: 	       return 0;
cannam@95:      }
cannam@95:      return 1;
cannam@95: }
cannam@95: 
cannam@95: static void dimcpy(bench_iodim *dst, const bench_iodim *src, int rnk)
cannam@95: {
cannam@95:      int i;
cannam@95:      if (FINITE_RNK(rnk))
cannam@95:           for (i = 0; i < rnk; ++i)
cannam@95:                dst[i] = src[i];
cannam@95: }
cannam@95: 
cannam@95: bench_tensor *tensor_append(const bench_tensor *a, const bench_tensor *b)
cannam@95: {
cannam@95:      if (!FINITE_RNK(a->rnk) || !FINITE_RNK(b->rnk)) {
cannam@95:           return mktensor(RNK_MINFTY);
cannam@95:      } else {
cannam@95: 	  bench_tensor *x = mktensor(a->rnk + b->rnk);
cannam@95:           dimcpy(x->dims, a->dims, a->rnk);
cannam@95:           dimcpy(x->dims + a->rnk, b->dims, b->rnk);
cannam@95: 	  return x;
cannam@95:      }
cannam@95: }
cannam@95: 
cannam@95: static int imax(int a, int b)
cannam@95: {
cannam@95:      return (a > b) ? a : b;
cannam@95: }
cannam@95: 
cannam@95: static int imin(int a, int b)
cannam@95: {
cannam@95:      return (a < b) ? a : b;
cannam@95: }
cannam@95: 
cannam@95: #define DEFBOUNDS(name, xs)			\
cannam@95: void name(bench_tensor *t, int *lbp, int *ubp)	\
cannam@95: {						\
cannam@95:      int lb = 0;				\
cannam@95:      int ub = 1;				\
cannam@95:      int i;					\
cannam@95: 						\
cannam@95:      BENCH_ASSERT(FINITE_RNK(t->rnk));		\
cannam@95: 						\
cannam@95:      for (i = 0; i < t->rnk; ++i) {		\
cannam@95: 	  bench_iodim *d = t->dims + i;		\
cannam@95: 	  int n = d->n;				\
cannam@95: 	  int s = d->xs;			\
cannam@95: 	  lb = imin(lb, lb + s * (n - 1));	\
cannam@95: 	  ub = imax(ub, ub + s * (n - 1));	\
cannam@95:      }						\
cannam@95: 						\
cannam@95:      *lbp = lb;					\
cannam@95:      *ubp = ub;					\
cannam@95: }
cannam@95: 
cannam@95: DEFBOUNDS(tensor_ibounds, is)
cannam@95: DEFBOUNDS(tensor_obounds, os)
cannam@95: 
cannam@95: bench_tensor *tensor_copy(const bench_tensor *sz)
cannam@95: {
cannam@95:      bench_tensor *x = mktensor(sz->rnk);
cannam@95:      dimcpy(x->dims, sz->dims, sz->rnk);
cannam@95:      return x;
cannam@95: }
cannam@95: 
cannam@95: /* Like tensor_copy, but copy only rnk dimensions starting with start_dim. */
cannam@95: bench_tensor *tensor_copy_sub(const bench_tensor *sz, int start_dim, int rnk)
cannam@95: {
cannam@95:      bench_tensor *x;
cannam@95: 
cannam@95:      BENCH_ASSERT(FINITE_RNK(sz->rnk) && start_dim + rnk <= sz->rnk);
cannam@95:      x = mktensor(rnk);
cannam@95:      dimcpy(x->dims, sz->dims + start_dim, rnk);
cannam@95:      return x;
cannam@95: }
cannam@95: 
cannam@95: bench_tensor *tensor_copy_swapio(const bench_tensor *sz)
cannam@95: {
cannam@95:      bench_tensor *x = tensor_copy(sz);
cannam@95:      int i;
cannam@95:      if (FINITE_RNK(x->rnk))
cannam@95: 	  for (i = 0; i < x->rnk; ++i) {
cannam@95: 	       int s;
cannam@95: 	       s = x->dims[i].is;
cannam@95: 	       x->dims[i].is = x->dims[i].os;
cannam@95: 	       x->dims[i].os = s;
cannam@95: 	  }
cannam@95:      return x;
cannam@95: }