Add comments

taichi-dev · Sep 13, 2022 · c1f1674 · c1f1674
1 parent 1430a9f
commit c1f1674
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/taichi/codegen/cuda/codegen_cuda.cpp b/taichi/codegen/cuda/codegen_cuda.cpp
@@ -663,6 +663,8 @@ class TaskCodeGenCUDA : public TaskCodeGenLLVM {
         TI_NOT_IMPLEMENTED
       }
     } else {
+      // Note that ret_type here cannot be integral because pow with an
+      // integral exponent has been demoted in the demote_operations pass
       if (ret_type->is_primitive(PrimitiveTypeID::f32)) {
         llvm_val[stmt] = create_call("__nv_powf", {lhs, rhs});
       } else if (ret_type->is_primitive(PrimitiveTypeID::f64)) {

diff --git a/taichi/codegen/llvm/codegen_llvm.cpp b/taichi/codegen/llvm/codegen_llvm.cpp
@@ -658,6 +658,8 @@ void TaskCodeGenLLVM::visit(BinaryOpStmt *stmt) {
       }
     } else if (op == BinaryOpType::pow) {
       if (arch_is_cpu(current_arch())) {
+        // Note that ret_type here cannot be integral because pow with an
+        // integral exponent has been demoted in the demote_operations pass
         if (ret_type->is_primitive(PrimitiveTypeID::f32)) {
           llvm_val[stmt] = create_call("pow_f32", {lhs, rhs});
         } else if (ret_type->is_primitive(PrimitiveTypeID::f64)) {