Cherry-pick SPIR-V assembly test improvements