mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
14 lines
193 B
Bash
14 lines
193 B
Bash
#!/bin/bash
|
|
|
|
arch=`__nvcc_device_query`
|
|
case "$arch" in
|
|
"90")
|
|
echo "90a" ;;
|
|
"100")
|
|
echo "100a" ;;
|
|
"121")
|
|
echo "121a" ;;
|
|
*)
|
|
echo "native" ;;
|
|
esac
|