[Bug] Fix spark unit test failures for test_add_rmsnorm_fp4_quant_cute_dsl#2573

kahyunnam merged 2 commits intoflashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

kahyunnam:knam/spark-unit-test-fixkahyunnam/flashinfer:knam/spark-unit-test-fixCopy head branch name to clipboard