mirrors/ZLUDA
0
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-08-03 06:40:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Support bra instruction

Browse source
This commit is contained in:
Andrzej Janik 2024-10-04 21:52:13 +02:00
commit 84364fa412
1 changed files with 23 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

26
ptx/src/pass/emit_llvm.rs
View file

@ -277,6 +277,11 @@ impl<'a, 'input> ModuleEmitContext<'a, 'input> {
unsafe { LLVMAppendBasicBlockInContext(self.context, fn_, LLVM_UNNAMED.as_ptr()) }; unsafe { LLVMAppendBasicBlockInContext(self.context, fn_, LLVM_UNNAMED.as_ptr()) };
unsafe { LLVMPositionBuilderAtEnd(self.builder.get(), real_bb) }; unsafe { LLVMPositionBuilderAtEnd(self.builder.get(), real_bb) };
let mut method_emitter = MethodEmitContext::new(self, fn_, variables_builder); let mut method_emitter = MethodEmitContext::new(self, fn_, variables_builder);
for statement in statements.iter() {
if let Statement::Label(label) = statement {
method_emitter.emit_label_initial(*label);
}
}
for statement in statements { for statement in statements {
method_emitter.emit_statement(statement)?; method_emitter.emit_statement(statement)?;
} }
@ -370,7 +375,7 @@ impl<'a, 'input> MethodEmitContext<'a, 'input> {
) -> Result<(), TranslateError> { ) -> Result<(), TranslateError> {
Ok(match statement { Ok(match statement {
Statement::Variable(var) => self.emit_variable(var)?, Statement::Variable(var) => self.emit_variable(var)?,
Statement::Label(label) => self.emit_label(label), Statement::Label(label) => self.emit_label_delayed(label)?,
Statement::Instruction(inst) => self.emit_instruction(inst)?, Statement::Instruction(inst) => self.emit_instruction(inst)?,
Statement::Conditional(_) => todo!(), Statement::Conditional(_) => todo!(),
Statement::LoadVar(var) => self.emit_load_variable(var)?, Statement::LoadVar(var) => self.emit_load_variable(var)?,
@ -404,7 +409,7 @@ impl<'a, 'input> MethodEmitContext<'a, 'input> {
Ok(()) Ok(())
} }
fn emit_label(&mut self, label: SpirvWord) { fn emit_label_initial(&mut self, label: SpirvWord) {
let block = unsafe { let block = unsafe {
LLVMAppendBasicBlockInContext( LLVMAppendBasicBlockInContext(
self.context, self.context,
@ -412,11 +417,19 @@ impl<'a, 'input> MethodEmitContext<'a, 'input> {
self.resolver.get_or_add_raw(label), self.resolver.get_or_add_raw(label),
) )
}; };
self.resolver
.register(label, unsafe { LLVMBasicBlockAsValue(block) });
}
fn emit_label_delayed(&mut self, label: SpirvWord) -> Result<(), TranslateError> {
let block = self.resolver.value(label)?;
let block = unsafe { LLVMValueAsBasicBlock(block) };
let last_block = unsafe { LLVMGetInsertBlock(self.builder) }; let last_block = unsafe { LLVMGetInsertBlock(self.builder) };
if unsafe { LLVMGetBasicBlockTerminator(last_block) } == ptr::null_mut() { if unsafe { LLVMGetBasicBlockTerminator(last_block) } == ptr::null_mut() {
unsafe { LLVMBuildBr(self.builder, block) }; unsafe { LLVMBuildBr(self.builder, block) };
} }
unsafe { LLVMPositionBuilderAtEnd(self.builder, block) }; unsafe { LLVMPositionBuilderAtEnd(self.builder, block) };
Ok(())
} }
fn emit_store_var(&mut self, store: StoreVarDetails) -> Result<(), TranslateError> { fn emit_store_var(&mut self, store: StoreVarDetails) -> Result<(), TranslateError> {
@ -441,7 +454,7 @@ impl<'a, 'input> MethodEmitContext<'a, 'input> {
ast::Instruction::Not { data, arguments } => todo!(), ast::Instruction::Not { data, arguments } => todo!(),
ast::Instruction::Or { data, arguments } => todo!(), ast::Instruction::Or { data, arguments } => todo!(),
ast::Instruction::And { data, arguments } => self.emit_and(arguments), ast::Instruction::And { data, arguments } => self.emit_and(arguments),
ast::Instruction::Bra { arguments } => todo!(), ast::Instruction::Bra { arguments } => self.emit_bra(arguments),
ast::Instruction::Call { data, arguments } => self.emit_call(data, arguments), ast::Instruction::Call { data, arguments } => self.emit_call(data, arguments),
ast::Instruction::Cvt { data, arguments } => todo!(), ast::Instruction::Cvt { data, arguments } => todo!(),
ast::Instruction::Shr { data, arguments } => todo!(), ast::Instruction::Shr { data, arguments } => todo!(),
@ -755,6 +768,13 @@ impl<'a, 'input> MethodEmitContext<'a, 'input> {
}); });
Ok(()) Ok(())
} }
fn emit_bra(&self, arguments: ptx_parser::BraArgs<SpirvWord>) -> Result<(), TranslateError> {
let src = self.resolver.value(arguments.src)?;
let src = unsafe { LLVMValueAsBasicBlock(src) };
unsafe { LLVMBuildBr(self.builder, src) };
Ok(())
}
} }
fn get_pointer_type<'ctx>( fn get_pointer_type<'ctx>(