Initial working PTX codegen