Skip to content
Snippets Groups Projects

GPU backend

Merged prathi3 requested to merge gpu-cg into main
7 files
+ 264
97
Compare changes
  • Side-by-side
  • Inline
Files
7
+ 2
0
@@ -11,6 +11,8 @@ pub fn device_placement(functions: &Vec<Function>, callgraph: &CallGraph) -> Vec
let mut devices = vec![];
for (idx, function) in functions.into_iter().enumerate() {
devices.push(Device::CUDA);
continue;
if let Some(device) = function.device {
devices.push(device);
} else if function.entry || callgraph.num_callees(FunctionID::new(idx)) != 0 {
Loading