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:
committed by
A. Unique TensorFlower
parent
dae0ae6879
commit
b65c8bb5d6
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user