[release/2.5] Update CUDAPluggableAllocator.h

[amd-sriram]

Altering the flag to use the correct streamType for CUDAPluggableAllocator. This is impacting Distributed Fused Adam in Rocm/APEX.

To fix the following error when building apex:
2025-03-25T14:19:25.0187731Z #5 1815.5 FAILED: /root/dockerbuild/pytorch.1.7.0a0/apex/build/temp.linux-x86_64-cpython-312/apex/contrib/csrc/nccl_allocator/NCCLAllocator_hip.o

torch/include/torch/csrc/cuda/CUDAPluggableAllocator.h:125:8: error: ‘void torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator::recordStream(const c10::DataPtr&, torch::cuda::CUDAPluggableAllocator::streamType)’ marked ‘override’, but does not override
2025-03-25T14:19:25.0192670Z #5 1815.5   125 |   void recordStream(const c10::DataPtr&, streamType stream) override;
2025-03-25T14:19:25.0192737Z #5 1815.5       |        ^~~~~~~~~~~~

Apex Commit : ROCm/apex#188

[rocm-repo-management-api]

Jenkins build for 5743a0cc85f7de5c8d42b14d28bc1941dfeb875c commit finished as FAILURE
Links: Blue Ocean view / Build artifacts

[jithunnair-amd]

This makes sense to me since when Apex (or any extension) includes the header, TORCH_HIP_VERSION is not defined, but USE_ROCM is.