Method CallCustomAnyHash

include/tvm/ffi/any.h:660–692

! * \brief Call the custom any hash function registered in type attribute column. * \param custom_any_hash The custom any hash function object or function pointer. * \param src The source Any. * \return The hash value. */

   * \return The hash value.
   */
  static uint64_t CallCustomAnyHash(const TVMFFIAny& custom_any_hash, const Any& src) {
    // NOTE: we explicitly use low-level ABI here since we do not want to have dep on function.h
    // it also keeps the logic simple
    if (custom_any_hash.type_index == TypeIndex::kTVMFFIOpaquePtr) {
      // we allow this attribute to be a function pointer for fast path
      using FCustomAnyHashPtr = int64_t (*)(const Any&);
      FCustomAnyHashPtr hash_func = reinterpret_cast<FCustomAnyHashPtr>(custom_any_hash.v_ptr);
      TVM_FFI_ICHECK_NOTNULL(hash_func);
      return static_cast<uint64_t>(
          (*hash_func)(src));  // NOLINT(clang-analyzer-core.CallAndMessage)
    } else {
      // alternatively it can be a ffi.Function object.
      TVM_FFI_ICHECK_EQ(custom_any_hash.type_index, TypeIndex::kTVMFFIFunction);
      TVMFFIAny arg = src.data_;
      TVMFFIAny result;
      result.type_index = TypeIndex::kTVMFFINone;
      result.zero_padding = 0;
      result.v_int64 = 0;
      TVMFFIFunctionCell* func_cell = TVMFFIFunctionGetCellPtr(custom_any_hash.v_obj);
      if (func_cell->cpp_call != nullptr) {
        // Fast path: invoke C++ ABI call directly when available.
        using FCppCall = void (*)(const void*, const TVMFFIAny*, int32_t, TVMFFIAny*);
        reinterpret_cast<FCppCall>(func_cell->cpp_call)(custom_any_hash.v_obj, &arg, 1, &result);
      } else {
        if (func_cell->safe_call(custom_any_hash.v_obj, &arg, 1, &result) != 0) {
          throw details::MoveFromSafeCallRaised();
        }
      }
      Any result_any = details::AnyUnsafe::MoveTVMFFIAnyToAny(&result);
      TVM_FFI_ICHECK_EQ(result_any.type_index(), TypeIndex::kTVMFFIInt);
      return static_cast<uint64_t>(result_any.data_.v_int64);
    }
  }
};

/*! \brief String-aware Any hash functor */