Debug: Also update testColumnParallel to use assert_close
testColumnParallel[True-2] failed in run 2 - update it to use
torch.testing.assert_close for better error messages showing actual
numerical differences.
Signed-off-by: Masahiro Tanaka <mtanaka@anyscale.com>
