nvptx 0.2.3

Rust to PTX compiler using LLVM/NVPTX target
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use failure;
use std::{io, process};

#[derive(Debug, Clone, Copy)]
pub enum Step {
    Install,
    Ready,
    Link,
    Build,
    Convert,
    Load,
}

#[derive(Fail, Debug)]
pub enum CompileError {
    #[fail(
        display = "External command {} failed during {:?} step. Return code: {}",
        command,
        step,
        error_code
    )]
    CommandFailure {
        step: Step,
        command: String,
        error_code: i32,
    },

    #[fail(
        display = "External command {} failed during {:?}. Please ensure it is installed.",
        command,
        step,
    )]
    CommandIOFailure {
        step: Step,
        command: String,
        error: io::Error,
    },

    #[fail(
        display = "Error during {:?} step: {:?}, error: {:?}",
        step,
        comment,
        error
    )]
    OtherError {
        step: Step,
        comment: String,
        error: failure::Error,
    },
}

pub fn err_msg(step: Step, comment: &str) -> CompileError {
    CompileError::OtherError {
        step,
        comment: comment.to_owned(),
        error: failure::err_msg(comment.to_owned()),
    }
}

pub type Result<T> = ::std::result::Result<T, CompileError>;
pub type ResultAny<T> = ::std::result::Result<T, failure::Error>;

pub trait Logging {
    type T;
    fn log_unwrap(self, step: Step) -> Result<Self::T>;
    fn log(self, step: Step, comment: &str) -> Result<Self::T>;
}

impl<T, E: Into<failure::Error>> Logging for ::std::result::Result<T, E> {
    type T = T;

    fn log_unwrap(self, step: Step) -> Result<Self::T> {
        self.log(step, "Unknown IO Error")
    }

    fn log(self, step: Step, comment: &str) -> Result<Self::T> {
        self.map_err(|e| {
            CompileError::OtherError {
                step,
                comment: comment.to_owned(),
                error: e.into(),
            }.into()
        })
    }
}

pub trait CheckRun {
    fn check_run(&mut self, step: Step) -> Result<()>;
}

impl CheckRun for process::Command {
    fn check_run(&mut self, step: Step) -> Result<()> {
        let st = self.status().map_err(|error| {
            let command = format!("{:?}", self);
            CompileError::CommandIOFailure {
                step,
                command,
                error,
            }
        })?;
        match st.code() {
            Some(error_code) => {
                if error_code != 0 {
                    let command = format!("{:?}", self);
                    Err(CompileError::CommandFailure {
                        step,
                        command,
                        error_code,
                    }.into())
                } else {
                    Ok(())
                }
            }
            None => Ok(()),
        }
    }
}