NUS-HPC-AI-Lab/VideoSys

Fails for batch size other than 1

pzpzpzp2 opened this issue 10 months ago · 2 comments

pzpzpzp2 commented 10 months ago

https://github.com/NUS-HPC-AI-Lab/OpenDiT/blob/9ca54720f3aa4c3a20c642059be63d61308b1bf1/tests/test_fused_modulate.py#L7
fused_modulate does not work for batch sizes other than 1

oahzxl commented 10 months ago

get it. will look into this problem. thanks for your feedback

oahzxl commented 4 months ago

close due to outdated