shtinsa commented on code in PR #11341:
URL: https://github.com/apache/tvm/pull/11341#discussion_r885601380
##########
tests/python/unittest/test_micro_model_library_format.py:
##########
@@ -418,14 +419,24 @@ def test_export_byoc_c_module():
with tf.extractfile("./metadata.json") as f:
metadata = json.load(f)
main_md = metadata["memory"]["functions"]["main"]
- assert main_md == [
- {
- "constants_size_bytes": 0,
- "device": 1,
- "io_size_bytes": 4800,
- "workspace_size_bytes": 800,
- }
- ]
+ if platform.architecture()[0] == "64bit":
+ assert main_md == [
+ {
+ "constants_size_bytes": 0,
+ "device": 1,
+ "io_size_bytes": 4800,
+ "workspace_size_bytes": 1264,
+ }
+ ]
+ else:
+ assert main_md == [
+ {
+ "constants_size_bytes": 0,
+ "device": 1,
+ "io_size_bytes": 4800,
+ "workspace_size_bytes": 1248,
+ }
Review Comment:
The concat code has 2 arrays and access to these arrays require 8 bytes for
x64 and 4 bytes for 32 bits systems. That should be reason of difference.
Should I recheck it?
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
