aoti diag: cuda geometry
Browse files
app.py
CHANGED
|
@@ -456,7 +456,7 @@ def aoti_diag() -> str:
|
|
| 456 |
|
| 457 |
import vsa_h3
|
| 458 |
|
| 459 |
-
geometry = vsa_h3.geometry_from_layout(tags, position_ids, 0.9)
|
| 460 |
if geometry is None:
|
| 461 |
return "**FAILED**: synthetic bench layout did not yield VSA geometry."
|
| 462 |
|
|
|
|
| 456 |
|
| 457 |
import vsa_h3
|
| 458 |
|
| 459 |
+
geometry = vsa_h3.geometry_from_layout(tags.to(device), position_ids.to(device), 0.9)
|
| 460 |
if geometry is None:
|
| 461 |
return "**FAILED**: synthetic bench layout did not yield VSA geometry."
|
| 462 |
|