mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
b6852778ff
Add magma build for CUDA 13.0 after almalinux docker is available https://github.com/pytorch/pytorch/issues/159779 Pull Request resolved: https://github.com/pytorch/pytorch/pull/160770 Approved by: https://github.com/atalman Co-authored-by: Andrey Talman <atalman@fb.com> Co-authored-by: Wei Wang <weiwan@nvidia.com>
27 lines
999 B
Diff
27 lines
999 B
Diff
diff --git a/interface_cuda/interface.cpp b/interface_cuda/interface.cpp
|
|
index 73fed1b20..e77519bfe 100644
|
|
--- a/interface_cuda/interface.cpp
|
|
+++ b/interface_cuda/interface.cpp
|
|
@@ -438,14 +438,20 @@ magma_print_environment()
|
|
cudaDeviceProp prop;
|
|
err = cudaGetDeviceProperties( &prop, dev );
|
|
check_error( err );
|
|
+ #ifdef MAGMA_HAVE_CUDA
|
|
+#if CUDA_VERSION < 13000
|
|
printf( "%% device %d: %s, %.1f MHz clock, %.1f MiB memory, capability %d.%d\n",
|
|
dev,
|
|
prop.name,
|
|
prop.clockRate / 1000.,
|
|
+#else
|
|
+ printf( "%% device %d: %s, ??? MHz clock, %.1f MiB memory, capability %d.%d\n",
|
|
+ dev,
|
|
+ prop.name,
|
|
+#endif
|
|
prop.totalGlobalMem / (1024.*1024.),
|
|
prop.major,
|
|
prop.minor );
|
|
- #ifdef MAGMA_HAVE_CUDA
|
|
int arch = prop.major*100 + prop.minor*10;
|
|
if ( arch < MAGMA_CUDA_ARCH_MIN ) {
|
|
printf("\n"
|