jbuckman

Ah, that was exactly it -- thank you!

That other issue is not solved by the Triton nightly (which, as per their instructions, I got from here: `pip install -U --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/ triton-nightly`), because the Triton nightly build...

This code is from a much older version of DyNet, so there are probably several similar compatibility issues by now, sorry! I will try to update it in the next...