You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2023/01/04 10:45:44 UTC

[GitHub] [tvm] lightzhan-intellif opened a new issue, #13695: [Bug] [TVMScript] Weird output of my script

lightzhan-intellif opened a new issue, #13695:
URL: https://github.com/apache/tvm/issues/13695

   Hi, all
   I have a program like this:
   ```python
   import numpy as np
   from tvm.script import tir as T
   import tvm
   
   @T.prim_func
   def test_case(A: T.Buffer[(1, ), "uint8"], B: T.Buffer[(1, ), "float32"]):
     B[0] = A[0]
   
   
   if __name__ == "__main__":
     compiled = tvm.build(test_case, target="llvm")
     in_data = np.empty((1,), dtype=np.uint8)
     in_data[0] = 100
     out_data = np.empty((1,), dtype=np.float32)
     A_tvm = tvm.nd.array(in_data, tvm.cpu(0))
     B_tvm = tvm.nd.array(out_data, tvm.cpu(0))
     compiled(A_tvm, B_tvm)
     print(B_tvm.numpy())
   ```
   I just want to assign a value `100` typed uint8 to an address of type float. In python, we should get the result `100` in the B[0], but it give me a weird result `[2.0000238]`.
   
   I change the target to `c`, and get the following code:
   ```c
   // tvm target: c -keys=cpu 
   #define TVM_EXPORTS
   #include "tvm/runtime/c_runtime_api.h"
   #include "tvm/runtime/c_backend_api.h"
   #include <math.h>
   #ifdef __cplusplus
   extern "C"
   #endif
   TVM_DLL int32_t default_function(void* args, int32_t* arg_type_ids, int32_t num_args, void* out_ret_value, int32_t* out_ret_tcode, void* resource_handle) {
     void* arg_A_handle = (((TVMValue*)args)[0].v_handle);
     int32_t arg_A_handle_code = arg_type_ids[0];
     void* arg_B_handle = (((TVMValue*)args)[1].v_handle);
     int32_t arg_B_handle_code = arg_type_ids[1];
     void* A = (((DLTensor*)arg_A_handle)[0].data);
     void* arg_A_handle_shape = (((DLTensor*)arg_A_handle)[0].shape);
     void* arg_A_handle_strides = (((DLTensor*)arg_A_handle)[0].strides);
     int32_t dev_id = (((DLTensor*)arg_A_handle)[0].device.device_id);
     void* B = (((DLTensor*)arg_B_handle)[0].data);
     void* arg_B_handle_shape = (((DLTensor*)arg_B_handle)[0].shape);
     void* arg_B_handle_strides = (((DLTensor*)arg_B_handle)[0].strides);
     if (!(arg_A_handle_strides == NULL)) {
     }
     if (!(arg_B_handle_strides == NULL)) {
     }
     *(uint8_t*)(((float*)B) + 0) = ((uint8_t*)A)[0];
     return 0;
   }
   
   // CodegenC: NOTE: Auto-generated entry function
   #ifdef __cplusplus
   extern "C"
   #endif
   TVM_DLL int32_t __tvm_main__(void* args, int* arg_type_ids, int num_args, void* out_ret_value, int* out_ret_tcode, void* resource_handle) {
     return default_function(args, arg_type_ids, num_args, out_ret_value, out_ret_tcode, resource_handle);
   }
   ```
   From the code , you can see that tvm does not do any casting to the assignment operation.


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org