15 |
|
|
16 |
structure IL = HighIL |
structure IL = HighIL |
17 |
structure Op = HighOps |
structure Op = HighOps |
18 |
|
structure Ty = Types |
19 |
|
structure TU = TypeUtil |
20 |
|
structure MV = MetaVar |
21 |
|
|
22 |
fun simpleOP rator (y, [], xs) = [IL.ASSIGN(y, IL.OP(rator, xs))] |
fun assign (y, rator, xs) = [IL.ASSIGN(y, IL.OP(rator, xs))] |
23 |
|
|
24 |
|
fun simpleOP rator (y, [], xs) = assign (y, rator, xs) |
25 |
|
|
26 |
|
fun tensorOP rator (y, [sv], xs) = (case TU.prunShape(MV.toShape sv) |
27 |
|
of Ty.Shape dd => assign (y, IP.OP(rator(Op.TensorTy dd), xs)) |
28 |
|
| shp => raise Fail("unresolved shape " ^ TU.shapeToString shp) |
29 |
|
(* end case *)) |
30 |
|
|
31 |
val tbl = let |
val tbl = let |
32 |
val tbl = Var.Tbl.mkTable (128, Fail "Tanslate table") |
val tbl = Var.Tbl.mkTable (128, Fail "Translate table") |
33 |
in |
in |
34 |
List.app (Var.Tbl.insert tbl) [ |
List.app (Var.Tbl.insert tbl) [ |
35 |
(add_ii, simpleOp(OP.Add OP.IntTy)), |
(add_ii, simpleOp(OP.Add OP.IntTy)), |
36 |
(add_tt, fn (y, [SK], xs) => ??), |
(add_tt, tensorOp OP.Add), |
37 |
(sub_ii, simpleOp(OP.Sub OP.IntTy)), |
(sub_ii, simpleOp(OP.Sub OP.IntTy)), |
38 |
(sub_tt, fn (y, [SK], xs) => ??), |
(sub_tt, tensorOp OP.Sub), |
39 |
(mul_ii, simpleOp(OP.Mul OP.IntTy)), |
(mul_ii, simpleOp(OP.Mul OP.IntTy)), |
40 |
(mul_rr, simpleOp(OP.Mul(OP.TensorTy[]))), |
(mul_rr, simpleOp(OP.Mul(OP.TensorTy[]))), |
41 |
(mul_rt, fn (y, [SK], xs) => ??), |
(mul_rt, tensorOp OP.Scale), |
42 |
(mul_tr, fn (y, [SK], xs) => ??), |
(mul_tr, fn (y, sv, [t, r]) => tensorOp OP.Scale (y, sv, [r, t])), |
43 |
(div_ii, simpleOp(OP.Div OP.IntTy)), |
(div_ii, simpleOp(OP.Div OP.IntTy)), |
44 |
(div_rr, simpleOp(OP.Div(OP.TensorTy[]))), |
(div_rr, simpleOp(OP.Div(OP.TensorTy[]))), |
45 |
(div_tr, fn (y, [SK], xs) => ??), |
(div_tr, fn (y, [SK], xs) => ??), (*FIXME*) |
46 |
(lt_ii, simpleOp(OP.LT OP.IntTy)), |
(lt_ii, simpleOp(OP.LT OP.IntTy)), |
47 |
(lt_rr, simpleOp(OP.LT(OP.TensorTy[]))), |
(lt_rr, simpleOp(OP.LT(OP.TensorTy[]))), |
48 |
(lte_ii, simpleOp(OP.LTE OP.IntTy)), |
(lte_ii, simpleOp(OP.LTE OP.IntTy)), |
60 |
(neq_ss, simpleOp(OP.NEQ OP.StringTy)), |
(neq_ss, simpleOp(OP.NEQ OP.StringTy)), |
61 |
(neq_rr, simpleOp(OP.NEQ(OP.TensorTy[]))), |
(neq_rr, simpleOp(OP.NEQ(OP.TensorTy[]))), |
62 |
(neg_i, simpleOp(OP.Neg OP.IntTy)), |
(neg_i, simpleOp(OP.Neg OP.IntTy)), |
63 |
(neg_t, fn (y, [SK], xs) => ??), |
(neg_t, tensorOp OP.Neg), |
64 |
(neg_f, fn (y, [DK, NK, SK], xs) => ??), |
(neg_f, fn (y, [DK, NK, SK], xs) => ??), (*FIXME*) |
65 |
(op_at, fn (y, [DK, NK, SK], xs) => ??), |
(op_at, fn (y, [DK, NK, SK], xs) => ??), (*FIXME*) |
66 |
(op_D, fn (y, [DK, NK, SK], xs) => ??), |
(op_D, fn (y, [DK, NK, SK], xs) => ??), (*FIXME*) |
67 |
(op_norm, fn (y, [SK], xs) => ??), |
(op_norm, tensorOp OP.Norm), |
68 |
(op_not, simpleOp OP.Not), |
(op_not, simpleOp OP.Not), |
69 |
(op_subscript, fn (y, [SK, NK], xs) => ??), |
(op_subscript, fn (y, [SK, NK], xs) => ??), (*FIXME*) |
70 |
(fn_CL, fn (y, [N3, N3], xs) => ??), |
(fn_CL, fn (y, [N3, N3], xs) => ??), (*FIXME*) |
71 |
(fn_convolve, fn (y, [DK, NK, SK], xs) => ??), |
(fn_convolve, fn (y, [DK, NK, SK], xs) => ??), (*FIXME*) |
72 |
(fn_cos, simpleOp OP.Cos), |
(fn_cos, simpleOp OP.Cos), |
73 |
(fn_dot, fn (y, [NK], xs) => ??), |
(fn_dot, fn (y, [dv], xs) => (case TU.pruneDim(MV.toDim dv) |
74 |
(fn_inside, fn (y, [DK, NK, SK], xs) => ??), |
of Ty.DimConst d => assign (y, OP.Dot(OP.TensorTy[d]), xs) |
75 |
(fn_load, fn (y, [NK, SK], xs) => ??), |
| dim => raise Fail("unresolved dimension "^TU.dimToString dim) |
76 |
|
(* end case *))), |
77 |
|
(fn_inside, fn (y, [DK, NK, SK], xs) => ??), (*FIXME*) |
78 |
|
(fn_load, fn (y, [NK, SK], xs) => ??), (*FIXME*) |
79 |
(fn_max, simpleOp Op.Min), |
(fn_max, simpleOp Op.Min), |
80 |
(fn_min, simpleOp OP.Max), |
(fn_min, simpleOp OP.Max), |
81 |
(fn_modulate, fn (y, [NK], xs) => ??), |
(fn_modulate, fn (y, [NK], xs) => ??), (*FIXME*) |
82 |
(fn_pow, simpleOp OP.Pow), |
(fn_pow, simpleOp OP.Pow), |
83 |
(fn_principleEvec, fn (y, [NK], xs) => ??), |
(fn_principleEvec, fn (y, [NK], xs) => ??), (*FIXME*) |
84 |
(fn_sin, simpleOp OP.Sin), |
(fn_sin, simpleOp OP.Sin), |
85 |
(kn_bspln3, fn (y, [], xs) => ??, |
(kn_bspln3, fn (y, [], xs) => ??, (*FIXME*) |
86 |
(kn_bspln5, fn (y, [], xs) => ??, |
(kn_bspln5, fn (y, [], xs) => ??, (*FIXME*) |
87 |
(kn_ctmr, fn (y, [], xs) => ??, |
(kn_ctmr, fn (y, [], xs) => ??, (*FIXME*) |
88 |
(kn_tent, fn (y, [], xs) => ??, |
(kn_tent, fn (y, [], xs) => ??, (*FIXME*) |
89 |
(i2r, fn (y, [], xs) => ??, |
(i2r, fn (y, [], xs) => ??, (*FIXME*) |
90 |
(input, fn (y, [TK], xs) => ??) |
(input, fn (y, [TK], xs) => ??) (*FIXME*) |
91 |
]; |
]; |
92 |
tbl |
tbl |
93 |
end |
end |