Enable UFMT on test/test_functionalization.py #5412

Job	Run time
docker-build (linux.12xlarge, pytorch-linux-focal-cuda12.1-cudnn8-py3-gcc9)	27m 17s
docker-build (linux.12xlarge, pytorch-linux-focal-cuda12.1-cudnn8-py3-gcc9-inductor-benchmarks)	29m 43s
docker-build (linux.12xlarge, pytorch-linux-focal-cuda11.8-cudnn8-py3-gcc9)	28m 27s
docker-build (linux.12xlarge, pytorch-linux-focal-py3.8-clang10)	17m 17s
docker-build (linux.12xlarge, pytorch-linux-focal-py3.11-clang10)	17m 57s
docker-build (linux.12xlarge, pytorch-linux-focal-py3.12-clang10)	13m 12s
docker-build (linux.12xlarge, pytorch-linux-focal-rocm-n-1-py3)	36m 32s
docker-build (linux.12xlarge, pytorch-linux-focal-rocm-n-py3)	40m 58s
docker-build (linux.12xlarge, pytorch-linux-jammy-cuda11.8-cudnn8-py3.8-clang12)	24m 0s
docker-build (linux.12xlarge, pytorch-linux-focal-py3-clang9-android-ndk-r21e)	13m 47s
docker-build (linux.12xlarge, pytorch-linux-jammy-py3.8-gcc11)	15m 54s
docker-build (linux.12xlarge, pytorch-linux-jammy-py3.8-gcc11-inductor-benchmarks)	18m 37s
docker-build (linux.12xlarge, pytorch-linux-jammy-xpu-2024.0-py3)	18m 7s
docker-build (linux.12xlarge, pytorch-linux-jammy-py3-clang15-asan)	8m 49s
docker-build (linux.12xlarge, pytorch-linux-focal-py3-clang10-onnx)	12m 33s
docker-build (linux.12xlarge, pytorch-linux-focal-linter)	9m 5s
docker-build (linux.12xlarge, pytorch-linux-jammy-cuda11.8-cudnn8-py3.9-linter)	17m 58s
docker-build (linux.12xlarge, pytorch-linux-jammy-py3-clang12-executorch)	12m 23s
docker-build (pytorch-linux-jammy-aarch64-py3.10-gcc11, linux.arm64.2xlarge)	25m 47s
	6h 28m 23s

Provide feedback