Enable CUTLASS grouped GEMM for pretraining wgrad on GB200 and H100 (resubmit) #654

Job	Run time
build_artifact (x86, linux.4xlarge, default, 3.9, gcc)	5m 51s
build_artifact (x86, linux.4xlarge, default, 3.9, clang)	5m 53s
build_artifact (x86, linux.4xlarge, default, 3.10, gcc)	5m 53s
build_artifact (x86, linux.4xlarge, default, 3.10, clang)	4m 44s
build_artifact (x86, linux.4xlarge, default, 3.11, gcc)	4m 38s
build_artifact (x86, linux.4xlarge, default, 3.11, clang)	4m 56s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.9, gcc)	3m 36s
build_artifact (x86, linux.4xlarge, default, 3.12, clang)	5m 25s
build_artifact (x86, linux.4xlarge, default, 3.13, gcc)	5m 10s
build_artifact (x86, linux.4xlarge, default, 3.12, gcc)	4m 19s
build_artifact (x86, linux.4xlarge, default, 3.13, clang)	4m 16s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.10, clang)	3m 35s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.12, clang)	4m 14s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.13, gcc)	4m 9s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.11, clang)	4m 7s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.13, clang)	3m 43s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.10, gcc)	3m 18s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.9, clang)	4m 11s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.11, gcc)	3m 30s
build_artifact (arm, linux.arm64.m7g.4xlarge, default, 3.12, gcc)	3m 22s
torchrec_cpu_tests	1s
	1h 28m 51s

Provide feedback