Commit 342dec5
authored
Create memerf with None stream for host data (NVIDIA#251)
Several tests creating memref for host data with stream are failing
NVIDIA@54f9819#diff-d10f6cc35783e94f23c4ff7a348efac1ede25554509e0631d82ba6701f838455R71.
It looks like MLIR-TRT is now propagating the stream correctly. Failure
is observed in following check:
```
if (type == PointerType::host) {
assert(alignment && !stream &&
"expected alignment, no stream for host allocation");
...
}
```
Tripy side fix is to pass None stream for host memref data.1 parent 59b9536 commit 342dec5
1 file changed
+2
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
28 | 28 | | |
29 | 29 | | |
30 | 30 | | |
| 31 | + | |
31 | 32 | | |
32 | 33 | | |
33 | 34 | | |
34 | 35 | | |
35 | 36 | | |
36 | | - | |
| 37 | + | |
37 | 38 | | |
38 | 39 | | |
39 | 40 | | |
| |||
0 commit comments