Diable L2_cahce_size display on ROCm platforms

This is not supported by clr and hsa and will return wrong results. Signed-off-by: Tim Gu <[email protected]>
ROCm · Jan 31, 2025 · dbeff96 · dbeff96
1 parent 6a28181
commit dbeff96
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch/csrc/cuda/Module.cpp b/torch/csrc/cuda/Module.cpp
@@ -1006,7 +1006,9 @@ static void registerCudaDeviceProperties(PyObject* module) {
 #ifndef FBCODE_CAFFE2
       .def_readonly("uuid", &cudaDeviceProp::uuid)
 #endif
+#if !USE_ROCM
       .def_readonly("L2_cache_size", &cudaDeviceProp::l2CacheSize)
+#endif
       .def("__repr__", [](const cudaDeviceProp& prop) {
         std::ostringstream stream;
         stream << "_CudaDeviceProperties(name='" << prop.name