Math Node 'MOD' WebGPU Error

[illestrater]

Description

Math Node 'MOD' produces error in WebGPU when input node is not a float, but works fine in WebGL

Reproduction steps

1. Create a 'mod' math node
2. Provide a TextureNode, rbga (or any other vector) as aNode
3. Provide a TimerNode (or any other float) as bNode
4. Attach mod node to material node's colorNode
5. WebGL runs fine, WebGPU throws error

Code

1 error(s) generated while compiling the shader:
:112:30 error: type mismatch for argument 1 in call to 'threejs_mod', expected 'f32', got 'vec4<f32>'
  DiffuseColor = threejs_mod( nodeVar2, NodeUniforms.nodeUniform2 );
                              ^^^^^^^^

Live example

Working WebGL with nodes: https://codesandbox.io/p/sandbox/three-js-webgpu-directional-light-forked-48y9yt?file=%2Fsrc%2Findex.mjs%3A40%2C17

The above will break using WebGPU renderer.

Screenshots

No response

Version

r160

Device

Desktop

Browser

Chrome

OS

MacOS