Skip to content

Commit 614d883

Browse files
author
Jessica Shi
committed
some tweaks
1 parent a73761c commit 614d883

File tree

2 files changed

+46
-20
lines changed

2 files changed

+46
-20
lines changed

documentation/docs/scheduling.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ Tensor<double> A("A", {512, 64}, csr);
77
Tensor<double> x("x", {64}, {Dense});
88
Tensor<double> y("y", {512}, {Dense});
99

10-
IndexVar i, j;
10+
IndexVar i("i"), j("j");
1111
y(i) = A(i, j) * x(j);
1212
IndexStmt stmt = y.getAssignment().concretize();
1313
```

javascripts/demo.js

Lines changed: 45 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -563,6 +563,18 @@ function demo() {
563563
};
564564

565565
var scheduleCommands = {
566+
pos: {
567+
parameters: ["Original IndexVar", "Derived IndexVar", "Accessed Tensor"],
568+
0: ["index dropdown", [1, "pos"]],
569+
1: ["default", ""],
570+
2: ["access dropdown"]
571+
},
572+
fuse: {
573+
parameters: ["Outer IndexVar", "Inner IndexVar", "Fused IndexVar"],
574+
0: ["index dropdown"],
575+
1: ["index dropdown"],
576+
2: ["default", "f"]
577+
},
566578
split: {
567579
parameters: ["Split IndexVar", "Outer IndexVar", "Inner IndexVar", "Split Factor"],
568580
0: ["index dropdown", [1, "0"], [2, "1"]],
@@ -577,29 +589,16 @@ function demo() {
577589
2: ["default", ""],
578590
3: ["number"]
579591
},
592+
precompute: {
593+
parameters: ["Original IndexVar", "Workspace IndexVar"],
594+
0: ["index dropdown", [1, ""]],
595+
1: ["default", ""]
596+
},
580597
reorder: {
581598
parameters: ["Reordered IndexVar", "Reordered IndexVar"],
582599
0: ["index dropdown"],
583600
1: ["index dropdown"]
584601
},
585-
pos: {
586-
parameters: ["Original IndexVar", "Derived IndexVar", "Accessed Tensor"],
587-
0: ["index dropdown", [1, "pos"]],
588-
1: ["default", ""],
589-
2: ["access dropdown"]
590-
},
591-
fuse: {
592-
parameters: ["Outer IndexVar", "Inner IndexVar", "Fused IndexVar"],
593-
0: ["index dropdown"],
594-
1: ["index dropdown"],
595-
2: ["default", "f"]
596-
},
597-
parallelize: {
598-
parameters: ["Parallel IndexVar", "Hardware", "Race Strategy"],
599-
0: ["index dropdown"],
600-
1: ["predefined dropdown", "CPU Thread", "Not Parallel", "Default Unit", "CPU Thread", "CPU Vector"],
601-
2: ["predefined dropdown", "No Races", "Ignore Races", "No Races", "Atomics", "Temporary", "Parallel Reduction"]
602-
},
603602
bound: {
604603
parameters: ["Original IndexVar", "Bounded IndexVar", "Bound", "Bound Type"],
605604
0: ["index dropdown", [1, "bound"]],
@@ -611,6 +610,12 @@ function demo() {
611610
parameters: ["Unrolled IndexVar", "Unroll Factor"],
612611
0: ["index dropdown"],
613612
1: ["number"]
613+
},
614+
parallelize: {
615+
parameters: ["Parallel IndexVar", "Hardware", "Race Strategy"],
616+
0: ["index dropdown"],
617+
1: ["predefined dropdown", "CPU Thread", "Not Parallel", "Default Unit", "CPU Thread", "CPU Vector"],
618+
2: ["predefined dropdown", "No Races", "Ignore Races", "No Races", "Atomics", "Temporary", "Parallel Reduction"]
614619
}
615620
};
616621

@@ -963,12 +968,15 @@ function demo() {
963968
tempCommand += param + "-";
964969
}
965970

971+
console.log(tempCommand);
972+
966973
if (valid) {
967974
// only add if user inputted all parameters
968975
command += tempCommand;
969976
}
970977
}
971978
command += "q";
979+
console.log(command);
972980

973981
var req = $.ajax({
974982
type: "POST",
@@ -1064,7 +1072,25 @@ function demo() {
10641072
C: { name: "Dense array", levels: { formats: ["d", "d"], ordering: [0, 1] } },
10651073
D: { name: "Dense array", levels: { formats: ["d", "d"], ordering: [0, 1] } },
10661074
},
1067-
schedule: []
1075+
schedule: [
1076+
{
1077+
command: "reorder",
1078+
numReordered: 4,
1079+
parameters: ["i", "k", "l", "j"]
1080+
},
1081+
{
1082+
command: "precompute",
1083+
parameters: ["j", "j"]
1084+
},
1085+
{
1086+
command: "split",
1087+
parameters: ["i", "i0", "i1", 32]
1088+
},
1089+
{
1090+
command: "parallelize",
1091+
parameters: ["i0", "CPU Thread", "No Races"]
1092+
}
1093+
]
10681094
}
10691095
};
10701096

0 commit comments

Comments
 (0)