Add EDSC support for loop.for operations

This CL adds support for loop.for operations in EDSC and adds a test.
This will be used in a followup commit to implement lowering of vector_transfer ops so that it works more generally and is not subject to affine constraints.

PiperOrigin-RevId: 275349796
This commit is contained in:
Nicolas Vasilache
2019-10-17 15:18:07 -07:00
committed by A. Unique TensorFlower
parent dae0ae6879
commit b65c8bb5d6
6 changed files with 135 additions and 39 deletions

View File

@@ -306,7 +306,7 @@ struct PythonLoopContext {
PythonValueHandle enter() {
ValueHandle iv(lb.value.getType());
builder = new LoopBuilder(&iv, lb.value, ub.value, step);
builder = new AffineLoopNestBuilder(&iv, lb.value, ub.value, step);
return iv;
}
@@ -318,7 +318,7 @@ struct PythonLoopContext {
PythonValueHandle lb, ub;
int64_t step;
LoopBuilder *builder = nullptr;
AffineLoopNestBuilder *builder = nullptr;
};
struct PythonLoopNestContext {