Fix memory size to MB

Author: mujacica

tests/test_integration_gpu.py

@@ -158,9 +158,7 @@ def test_gpu_context_structure_validation(cmake):
                 ), f"{context_key} memory_size should be an integer"
                 assert memory_size > 0, f"{context_key} memory_size should be positive"
                 # Should be at least 1MB (very conservative)
-                assert (
-                    memory_size >= 1024 * 1024
-                ), f"{context_key} memory size seems too small"
+                assert memory_size >= 1, f"{context_key} memory size seems too small"
 
 
 def test_gpu_context_cross_platform_compatibility(cmake):