cuda_std 0.2.2

Standard library for CUDA with rustc_codegen_nvvm
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

//! Utilities for configuring code based on the specified compute capability.

use cuda_std_macros::gpu_only;

#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
pub enum ComputeCapability {
    Compute35,
    Compute37,
    Compute50,
    Compute52,
    Compute53,
    Compute60,
    Compute61,
    Compute62,
    Compute70,
    Compute72,
    Compute75,
    Compute80,
}

impl ComputeCapability {
    /// Parses a compute capability from the `CUDA_ARCH` environment variable set by `cuda_builder`.
    /// This is a compile-time variable so any comparisons of the compute capability should expand to constant
    /// values.
    ///
    /// This allows you to use the current capability to decide what path to take in a function with the incorrect
    /// path being optimized away.
    #[gpu_only]
    #[inline(always)]
    pub fn from_cuda_arch_env() -> Self {
        // set by cuda_builder
        match env!("CUDA_ARCH") {
            "350" => ComputeCapability::Compute35,
            "370" => ComputeCapability::Compute37,
            "500" => ComputeCapability::Compute50,
            "520" => ComputeCapability::Compute52,
            "530" => ComputeCapability::Compute53,
            "600" => ComputeCapability::Compute60,
            "610" => ComputeCapability::Compute61,
            "620" => ComputeCapability::Compute62,
            "700" => ComputeCapability::Compute70,
            "720" => ComputeCapability::Compute72,
            "750" => ComputeCapability::Compute75,
            "800" => ComputeCapability::Compute80,
            _ => panic!("CUDA_ARCH had an invalid value"),
        }
    }
}