Mercurial > octave
annotate src/OPERATORS/op-m-cm.cc @ 9665:1dba57e9d08d
use blas_trans_type for xgemm
| author | Jaroslav Hajek <highegg@gmail.com> |
|---|---|
| date | Sat, 26 Sep 2009 10:41:07 +0200 |
| parents | 7e5b4de5fbfe |
| children | fb6b6fcafa62 |
| rev | line source |
|---|---|
| 2928 | 1 /* |
| 2 | |
| 7017 | 3 Copyright (C) 1996, 1997, 2000, 2001, 2002, 2003, 2004, 2005, 2006, |
| 8920 | 4 2007, 2008 John W. Eaton |
| 2928 | 5 |
| 6 This file is part of Octave. | |
| 7 | |
| 8 Octave is free software; you can redistribute it and/or modify it | |
| 9 under the terms of the GNU General Public License as published by the | |
| 7016 | 10 Free Software Foundation; either version 3 of the License, or (at your |
| 11 option) any later version. | |
| 2928 | 12 |
| 13 Octave is distributed in the hope that it will be useful, but WITHOUT | |
| 14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
| 15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
| 16 for more details. | |
| 17 | |
| 18 You should have received a copy of the GNU General Public License | |
| 7016 | 19 along with Octave; see the file COPYING. If not, see |
| 20 <http://www.gnu.org/licenses/>. | |
| 2928 | 21 |
| 22 */ | |
| 23 | |
| 24 #ifdef HAVE_CONFIG_H | |
| 25 #include <config.h> | |
| 26 #endif | |
| 27 | |
| 28 #include "mx-m-cm.h" | |
| 29 #include "mx-cm-m.h" | |
| 4544 | 30 #include "mx-nda-cnda.h" |
| 31 #include "mx-cnda-nda.h" | |
| 2928 | 32 |
| 33 #include "gripes.h" | |
| 4055 | 34 #include "oct-obj.h" |
| 2928 | 35 #include "ov.h" |
| 36 #include "ov-re-mat.h" | |
|
7789
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
37 #include "ov-flt-re-mat.h" |
| 2928 | 38 #include "ov-cx-mat.h" |
|
7789
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
39 #include "ov-flt-cx-mat.h" |
| 2928 | 40 #include "ov-typeinfo.h" |
| 41 #include "ops.h" | |
| 42 #include "xdiv.h" | |
| 43 #include "xpow.h" | |
| 44 | |
| 45 // matrix by complex matrix ops. | |
| 46 | |
| 4550 | 47 DEFNDBINOP_OP (add, matrix, complex_matrix, array, complex_array, +) |
| 48 DEFNDBINOP_OP (sub, matrix, complex_matrix, array, complex_array, -) | |
| 4543 | 49 |
| 2928 | 50 DEFBINOP_OP (mul, matrix, complex_matrix, *) |
| 5785 | 51 |
|
9663
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
52 DEFBINOP (trans_mul, matrix, complex_matrix) |
|
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
53 { |
|
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
54 CAST_BINOP_ARGS (const octave_matrix&, const octave_complex_matrix&); |
|
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
55 |
|
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
56 Matrix m1 = v1.matrix_value (); |
|
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
57 ComplexMatrix m2 = v2.complex_matrix_value (); |
|
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
58 |
|
9665
1dba57e9d08d
use blas_trans_type for xgemm
Jaroslav Hajek <highegg@gmail.com>
parents:
9663
diff
changeset
|
59 return ComplexMatrix (xgemm (m1, real (m2), blas_trans, blas_no_trans), |
|
1dba57e9d08d
use blas_trans_type for xgemm
Jaroslav Hajek <highegg@gmail.com>
parents:
9663
diff
changeset
|
60 xgemm (m1, imag (m2), blas_trans, blas_no_trans)); |
|
9663
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
61 } |
|
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
62 |
| 5785 | 63 DEFBINOP (div, matrix, complex_matrix) |
| 64 { | |
| 65 CAST_BINOP_ARGS (const octave_matrix&, const octave_complex_matrix&); | |
| 66 MatrixType typ = v2.matrix_type (); | |
| 67 | |
| 68 ComplexMatrix ret = xdiv (v1.matrix_value (), | |
| 69 v2.complex_matrix_value (), typ); | |
| 70 | |
| 71 v2.matrix_type (typ); | |
| 72 return ret; | |
| 73 } | |
| 2928 | 74 |
| 75 DEFBINOPX (pow, matrix, complex_matrix) | |
| 76 { | |
| 77 error ("can't do A ^ B for A and B both matrices"); | |
| 78 return octave_value (); | |
| 79 } | |
| 80 | |
| 5785 | 81 DEFBINOP (ldiv, matrix, complex_matrix) |
| 82 { | |
| 83 CAST_BINOP_ARGS (const octave_matrix&, const octave_complex_matrix&); | |
| 84 MatrixType typ = v1.matrix_type (); | |
| 85 | |
| 86 ComplexMatrix ret = xleftdiv (v1.matrix_value (), | |
| 87 v2.complex_matrix_value (), typ); | |
| 88 | |
| 89 v1.matrix_type (typ); | |
| 90 return ret; | |
| 91 } | |
| 2928 | 92 |
|
9662
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
93 DEFBINOP (trans_ldiv, matrix, complex_matrix) |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
94 { |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
95 CAST_BINOP_ARGS (const octave_matrix&, const octave_complex_matrix&); |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
96 MatrixType typ = v1.matrix_type (); |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
97 |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
98 ComplexMatrix ret = xleftdiv (v1.matrix_value (), |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
99 v2.complex_matrix_value (), typ, blas_trans); |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
100 |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
101 v1.matrix_type (typ); |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
102 return ret; |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
103 } |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
104 |
|
9588
319e2ab9b8ae
warn about the complex comparison ops
Jaroslav Hajek <highegg@gmail.com>
parents:
8920
diff
changeset
|
105 DEFNDCMPLXCMPOP_FN (lt, matrix, complex_matrix, array, complex_array, mx_el_lt) |
|
319e2ab9b8ae
warn about the complex comparison ops
Jaroslav Hajek <highegg@gmail.com>
parents:
8920
diff
changeset
|
106 DEFNDCMPLXCMPOP_FN (le, matrix, complex_matrix, array, complex_array, mx_el_le) |
|
319e2ab9b8ae
warn about the complex comparison ops
Jaroslav Hajek <highegg@gmail.com>
parents:
8920
diff
changeset
|
107 DEFNDCMPLXCMPOP_FN (eq, matrix, complex_matrix, array, complex_array, mx_el_eq) |
|
319e2ab9b8ae
warn about the complex comparison ops
Jaroslav Hajek <highegg@gmail.com>
parents:
8920
diff
changeset
|
108 DEFNDCMPLXCMPOP_FN (ge, matrix, complex_matrix, array, complex_array, mx_el_ge) |
|
319e2ab9b8ae
warn about the complex comparison ops
Jaroslav Hajek <highegg@gmail.com>
parents:
8920
diff
changeset
|
109 DEFNDCMPLXCMPOP_FN (gt, matrix, complex_matrix, array, complex_array, mx_el_gt) |
|
319e2ab9b8ae
warn about the complex comparison ops
Jaroslav Hajek <highegg@gmail.com>
parents:
8920
diff
changeset
|
110 DEFNDCMPLXCMPOP_FN (ne, matrix, complex_matrix, array, complex_array, mx_el_ne) |
| 2928 | 111 |
| 4550 | 112 DEFNDBINOP_FN (el_mul, matrix, complex_matrix, array, complex_array, product) |
| 113 DEFNDBINOP_FN (el_div, matrix, complex_matrix, array, complex_array, quotient) | |
| 114 DEFNDBINOP_FN (el_pow, matrix, complex_matrix, array, complex_array, elem_xpow) | |
| 2928 | 115 |
| 116 DEFBINOP (el_ldiv, matrix, complex_matrix) | |
| 117 { | |
| 118 CAST_BINOP_ARGS (const octave_matrix&, const octave_complex_matrix&); | |
| 119 | |
| 4550 | 120 return quotient (v2.complex_array_value (), v1.array_value ()); |
| 2928 | 121 } |
| 122 | |
| 4550 | 123 DEFNDBINOP_FN (el_and, matrix, complex_matrix, array, complex_array, mx_el_and) |
| 124 DEFNDBINOP_FN (el_or, matrix, complex_matrix, array, complex_array, mx_el_or) | |
| 2928 | 125 |
| 4915 | 126 DEFNDCATOP_FN (m_cm, matrix, complex_matrix, array, complex_array, concat) |
| 127 | |
| 2928 | 128 DEFCONV (complex_matrix_conv, matrix, complex_matrix) |
| 129 { | |
| 130 CAST_CONV_ARG (const octave_matrix&); | |
| 131 | |
| 4543 | 132 return new octave_complex_matrix (ComplexNDArray (v.array_value ())); |
| 2928 | 133 } |
| 134 | |
| 135 void | |
| 136 install_m_cm_ops (void) | |
| 137 { | |
| 3538 | 138 INSTALL_BINOP (op_add, octave_matrix, octave_complex_matrix, add); |
| 139 INSTALL_BINOP (op_sub, octave_matrix, octave_complex_matrix, sub); | |
| 140 INSTALL_BINOP (op_mul, octave_matrix, octave_complex_matrix, mul); | |
| 141 INSTALL_BINOP (op_div, octave_matrix, octave_complex_matrix, div); | |
| 142 INSTALL_BINOP (op_pow, octave_matrix, octave_complex_matrix, pow); | |
| 143 INSTALL_BINOP (op_ldiv, octave_matrix, octave_complex_matrix, ldiv); | |
| 144 INSTALL_BINOP (op_lt, octave_matrix, octave_complex_matrix, lt); | |
| 145 INSTALL_BINOP (op_le, octave_matrix, octave_complex_matrix, le); | |
| 146 INSTALL_BINOP (op_eq, octave_matrix, octave_complex_matrix, eq); | |
| 147 INSTALL_BINOP (op_ge, octave_matrix, octave_complex_matrix, ge); | |
| 148 INSTALL_BINOP (op_gt, octave_matrix, octave_complex_matrix, gt); | |
| 149 INSTALL_BINOP (op_ne, octave_matrix, octave_complex_matrix, ne); | |
| 150 INSTALL_BINOP (op_el_mul, octave_matrix, octave_complex_matrix, el_mul); | |
| 151 INSTALL_BINOP (op_el_div, octave_matrix, octave_complex_matrix, el_div); | |
| 152 INSTALL_BINOP (op_el_pow, octave_matrix, octave_complex_matrix, el_pow); | |
| 153 INSTALL_BINOP (op_el_ldiv, octave_matrix, octave_complex_matrix, el_ldiv); | |
| 154 INSTALL_BINOP (op_el_and, octave_matrix, octave_complex_matrix, el_and); | |
| 155 INSTALL_BINOP (op_el_or, octave_matrix, octave_complex_matrix, el_or); | |
|
9663
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
156 INSTALL_BINOP (op_trans_mul, octave_matrix, octave_complex_matrix, trans_mul); |
|
7e5b4de5fbfe
improve mixed real x complex ops
Jaroslav Hajek <highegg@gmail.com>
parents:
9662
diff
changeset
|
157 INSTALL_BINOP (op_herm_mul, octave_matrix, octave_complex_matrix, trans_mul); |
|
9662
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
158 INSTALL_BINOP (op_trans_ldiv, octave_matrix, octave_complex_matrix, trans_ldiv); |
|
0d3b248f4ab6
further improve mixed real-complex division
Jaroslav Hajek <highegg@gmail.com>
parents:
9588
diff
changeset
|
159 INSTALL_BINOP (op_herm_ldiv, octave_matrix, octave_complex_matrix, trans_ldiv); |
| 2928 | 160 |
| 4915 | 161 INSTALL_CATOP (octave_matrix, octave_complex_matrix, m_cm); |
| 162 | |
| 2928 | 163 INSTALL_ASSIGNCONV (octave_matrix, octave_complex_matrix, octave_complex_matrix); |
|
7789
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
164 INSTALL_ASSIGNCONV (octave_float_matrix, octave_complex_matrix, octave_float_complex_matrix); |
| 2928 | 165 |
| 3540 | 166 INSTALL_WIDENOP (octave_matrix, octave_complex_matrix, complex_matrix_conv); |
| 2928 | 167 } |
| 168 | |
| 169 /* | |
| 170 ;;; Local Variables: *** | |
| 171 ;;; mode: C++ *** | |
| 172 ;;; End: *** | |
| 173 */ |